test

yiyixuxu · yiyixuxu · commit ef7a14bea63c · 2025-02-13T06:15:22.000+01:00
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

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ def dummy_input(self):`
`51`	`51`	`"hidden_states": hidden_states,`
`52`	`52`	`"encoder_hidden_states": encoder_hidden_states,`
`53`	`53`	`"timestep": timestep,`
`54`		`- "attention_mask": attention_mask,`
	`54`	`+ "encoder_attention_mask": attention_mask,`
`55`	`55`	`}`
`56`	`56`
`57`	`57`	`@property`