model restored

Author: prishajain1

src/maxdiffusion/configs/base_wan_i2v_14b.yml

@@ -27,7 +27,7 @@ gcs_metrics: False
 save_config_to_gcs: False
 log_period: 100
 
-pretrained_model_name_or_path: 'Wan-AI/Wan2.1-I2V-14B-720P-Diffusers'
+pretrained_model_name_or_path: 'Wan-AI/Wan2.1-I2V-14B-480P-Diffusers'
 model_name: wan2.1
 model_type: 'I2V'
 

src/maxdiffusion/models/wan/transformers/transformer_wan.py

@@ -373,7 +373,6 @@ def __call__(
       rotary_emb: jax.Array,
       deterministic: bool = True,
       rngs: nnx.Rngs = None,
-      encoder_attention_mask: jax.Array = None,
   ):
     with self.conditional_named_scope("transformer_block"):
       shift_msa, scale_msa, gate_msa, c_shift_msa, c_scale_msa, c_gate_msa = jnp.split(