test

huggingface · Feb 13, 2025 · ef7a14b · ef7a14b
1 parent fc3aa8c
commit ef7a14b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/transformers/test_models_transformer_lumina2.py b/tests/models/transformers/test_models_transformer_lumina2.py
@@ -51,7 +51,7 @@ def dummy_input(self):
             "hidden_states": hidden_states,
             "encoder_hidden_states": encoder_hidden_states,
             "timestep": timestep,
-            "attention_mask": attention_mask,
+            "encoder_attention_mask": attention_mask,
         }
 
     @property