Trying text_mask 10

Author: prishajain1

src/maxdiffusion/pipelines/wan/wan_pipeline_i2v_2p2.py

@@ -273,7 +273,8 @@ def run_inference_2_2_i2v(
             encoder_attention_mask = jnp.concatenate([text_attention_mask, negative_text_attention_mask], axis=0)
         else:
             encoder_attention_mask = None
-        image_embeds = jnp.concatenate([image_embeds, image_embeds], axis=0)
+        if image_embeds is not None:
+            image_embeds = jnp.concatenate([image_embeds, image_embeds], axis=0)
         condition = jnp.concatenate([condition] * 2)
 
     def high_noise_branch(operands):