[diffusion] model: support qwen-image-edit-2511 (#15458)

2025-12-19 20:06:00 +08:00
parent 89512029f1
commit ba41080892
4 changed files with 52 additions and 7 deletions
@@ -20,6 +20,7 @@ class QwenImageArchConfig(DiTArchConfig):
    pooled_projection_dim: int = 768
    guidance_embeds: bool = False
    axes_dims_rope: Tuple[int, int, int] = (16, 56, 56)
+    zero_cond_t: bool = False

    stacked_params_mapping: list[tuple[str, str, str]] = field(
        default_factory=lambda: [
@@ -462,4 +462,5 @@ class QwenImageEditPlusPipelineConfig(QwenImageEditPipelineConfig):
        return {
            "txt_seq_lens": txt_seq_lens,
            "freqs_cis": ((img_cos, img_sin), (txt_cos, txt_sin)),
+            "img_shapes": img_shapes,
        }