Fix tests recurrent (#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]
This commit is contained in:
Pablo Montalvo 2024-08-13 23:40:50 +02:00 committed by GitHub
parent 9d2ab8824c
commit c1357834e8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -2829,8 +2829,8 @@ class ModelTesterMixin:
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()
)