From 6c7335e053ae882979c4fac01ad6866e564ac21f Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Mon, 22 Apr 2024 17:02:55 +0100 Subject: [PATCH] Jamba: fix left-padding test (#30389) fix test --- tests/models/jamba/test_modeling_jamba.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/jamba/test_modeling_jamba.py b/tests/models/jamba/test_modeling_jamba.py index f8e9fdb77b..ffe859bb59 100644 --- a/tests/models/jamba/test_modeling_jamba.py +++ b/tests/models/jamba/test_modeling_jamba.py @@ -483,7 +483,7 @@ class JambaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi 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()