[Revert] revert diffusion warmup (#382)

Signed-off-by: zjy0516 <riverclouds.zhu@qq.com>

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:

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,

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,