diff --git a/vllm_omni/diffusion/diffusion_engine.py b/vllm_omni/diffusion/diffusion_engine.py index ebd9ac5bd99..8d830255c99 100644 --- a/vllm_omni/diffusion/diffusion_engine.py +++ b/vllm_omni/diffusion/diffusion_engine.py @@ -32,12 +32,6 @@ def __init__(self, od_config: OmniDiffusionConfig): self._processes: list[mp.Process] = [] self._closed = False self._make_client() - try: - self._dummy_run() - except Exception as e: - logger.error(f"Dummy run failed: {e}") - self.close() - raise e def step(self, requests: list[OmniDiffusionRequest]): try: diff --git a/vllm_omni/diffusion/models/qwen_image/qwen_image_transformer.py b/vllm_omni/diffusion/models/qwen_image/qwen_image_transformer.py index b83731545fd..fee9f34edd4 100644 --- a/vllm_omni/diffusion/models/qwen_image/qwen_image_transformer.py +++ b/vllm_omni/diffusion/models/qwen_image/qwen_image_transformer.py @@ -303,7 +303,6 @@ def forward( return img_attn_output, txt_attn_output -@torch.compile(dynamic=True) class QwenImageTransformerBlock(nn.Module): def __init__( self, diff --git a/vllm_omni/diffusion/models/z_image/z_image_transformer.py b/vllm_omni/diffusion/models/z_image/z_image_transformer.py index 1250f460e5c..ddf3ce00d29 100644 --- a/vllm_omni/diffusion/models/z_image/z_image_transformer.py +++ b/vllm_omni/diffusion/models/z_image/z_image_transformer.py @@ -185,7 +185,6 @@ def forward(self, x): return self.w2(self.act(self.w13(x))) -@torch.compile(dynamic=True) class ZImageTransformerBlock(nn.Module): def __init__( self,