qwen_image: propagate attention mask. (#11966)

2026-01-23 03:02:31 +02:00
parent 09a2e67151
commit d7f3241bf6
2 changed files with 13 additions and 1 deletions
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -1578,6 +1578,9 @@ class QwenImage(BaseModel):

    def extra_conds(self, **kwargs):
        out = super().extra_conds(**kwargs)
+        attention_mask = kwargs.get("attention_mask", None)
+        if attention_mask is not None:
+            out['attention_mask'] = comfy.conds.CONDRegular(attention_mask)
        cross_attn = kwargs.get("cross_attn", None)
        if cross_attn is not None:
            out['c_crossattn'] = comfy.conds.CONDRegular(cross_attn)