[diffusion] improve: tiny speedup qwen-image-edit-2511 by avoiding unnecessary calculation (#15896)

python/sglang/multimodal_gen/configs/models/dits/qwenimage.py

@@ -38,8 +38,22 @@ class QwenImageArchConfig(DiTArchConfig):
         self.num_channels_latents = self.out_channels
 
 
+@dataclass
+class QwenImageEditPlus_2511_ArchConfig(DiTArchConfig):
+    zero_cond_t: bool = True
+
+
 @dataclass
 class QwenImageDitConfig(DiTConfig):
     arch_config: DiTArchConfig = field(default_factory=QwenImageArchConfig)
 
     prefix: str = "qwenimage"
+
+
+@dataclass
+class QwenImageEditPlus_2511_DitConfig(DiTConfig):
+    arch_config: DiTArchConfig = field(
+        default_factory=QwenImageEditPlus_2511_ArchConfig
+    )
+
+    prefix: str = "qwenimageedit"

python/sglang/multimodal_gen/configs/pipeline_configs/qwen_image.py

@@ -6,7 +6,10 @@ from typing import Callable
 import torch
 
 from sglang.multimodal_gen.configs.models import DiTConfig, EncoderConfig, VAEConfig
-from sglang.multimodal_gen.configs.models.dits.qwenimage import QwenImageDitConfig
+from sglang.multimodal_gen.configs.models.dits.qwenimage import (
+    QwenImageDitConfig,
+    QwenImageEditPlus_2511_DitConfig,
+)
 from sglang.multimodal_gen.configs.models.encoders.qwen_image import Qwen2_5VLConfig
 from sglang.multimodal_gen.configs.models.vaes.qwenimage import QwenImageVAEConfig
 from sglang.multimodal_gen.configs.pipeline_configs.base import (
@@ -415,7 +418,6 @@ class QwenImageEditPlusPipelineConfig(QwenImageEditPipelineConfig):
         assert batch_size == 1
         height = batch.height
         width = batch.width
-        image_size = batch.original_condition_image_size
 
         vae_scale_factor = self.get_vae_scale_factor()
 
@@ -474,6 +476,11 @@ class QwenImageEditPlusPipelineConfig(QwenImageEditPipelineConfig):
         }
 
 
+@dataclass
+class QwenImageEditPlus_2511_PipelineConfig(QwenImageEditPlusPipelineConfig):
+    dit_config: DiTConfig = field(default_factory=QwenImageEditPlus_2511_DitConfig)
+
+
 @dataclass
 class QwenImageLayeredPipelineConfig(QwenImageEditPipelineConfig):
     resolution: int = 640  # TODO: allow user to set resolution