diff --git a/vllm_omni/diffusion/diffusion_engine.py b/vllm_omni/diffusion/diffusion_engine.py index ebd9ac5bd..8d830255c 100644 --- a/vllm_omni/diffusion/diffusion_engine.py +++ b/vllm_omni/diffusion/diffusion_engine.py @@ -32,12 +32,6 @@ class DiffusionEngine: 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 b83731545..fee9f34ed 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 @@ class QwenImageCrossAttention(nn.Module): 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 1250f460e..ddf3ce00d 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 @@ class FeedForward(nn.Module): return self.w2(self.act(self.w13(x))) -@torch.compile(dynamic=True) class ZImageTransformerBlock(nn.Module): def __init__( self,