Update forward signature test for vision models (#27681)

* Update forward signature * Empty-Commit
2023-11-27 15:48:17 +01:00
parent 1d7f406e19
commit 59499bbe8b
48 changed files with 1 additions and 612 deletions
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -543,7 +543,7 @@ class ModelTesterMixin:
                )
                self.assertListEqual(arg_names[: len(expected_arg_names)], expected_arg_names)
            else:
-                expected_arg_names = ["input_ids"]
+                expected_arg_names = [model.main_input_name]
                self.assertListEqual(arg_names[:1], expected_arg_names)

    def check_training_gradient_checkpointing(self, gradient_checkpointing_kwargs=None):