Jamba: fix left-padding test (#30389)

fix test
huggingface · Apr 22, 2024 · 6c7335e · 6c7335e
1 parent f3b3533
commit 6c7335e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/jamba/test_modeling_jamba.py b/tests/models/jamba/test_modeling_jamba.py
@@ -483,7 +483,7 @@ def _prepare_model_kwargs(input_ids, attention_mask, signature):
             return model_kwargs
 
         for model_class in decoder_only_classes:
-            config, input_ids, attention_mask, _ = self._get_input_ids_and_config()
+            config, input_ids, attention_mask = self._get_input_ids_and_config()
             model = model_class(config).to(torch_device).eval()
             signature = inspect.signature(model.forward).parameters.keys()