Fix tests recurrent (huggingface#32651)

* add fix for recurrentgemma * [no-filter] * trigger-ci * [no-filter] * [no-filter] * attempt to fix mysterious zip error * [no-filter] * fix lookup error * [no-filter] * remove summarization hack * [no-filter]
clownrat6 · Aug 13, 2024 · c135783 · c135783
1 parent 9d2ab88
commit c135783
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py
@@ -2829,8 +2829,8 @@ def test_inputs_embeds_matches_input_ids_with_generate(self):
             model.eval()
 
             model_forward_args = inspect.signature(model.forward).parameters
-            if "inputs_embeds" not in model_forward_args:
-                self.skipTest(reason="This model doesn't use `inputs_embeds`")
+            if any(argument not in model_forward_args for argument in ["inputs_embeds", "position_ids"]):
+                self.skipTest(reason="This model doesn't use `inputs_embeds` or `position_ids`.")
             has_inputs_embeds_forwarding = "inputs_embeds" in set(
                 inspect.signature(model.prepare_inputs_for_generation).parameters.keys()
             )