[AutoBackbone] Add test (#26094)

* Add test * Add config_class
2023-09-18 23:47:54 +02:00
parent 97f439aed8
commit de8bec6df3
3 changed files with 14 additions and 2 deletions
--- a/tests/models/timm_backbone/test_modeling_timm_backbone.py
+++ b/tests/models/timm_backbone/test_modeling_timm_backbone.py
@@ -106,8 +106,9 @@ class TimmBackboneModelTest(ModelTesterMixin, BackboneTesterMixin, PipelineTeste
    has_attentions = False

    def setUp(self):
+        self.config_class = PretrainedConfig
        self.model_tester = TimmBackboneModelTester(self)
-        self.config_tester = ConfigTester(self, config_class=PretrainedConfig, has_text_modality=False)
+        self.config_tester = ConfigTester(self, config_class=self.config_class, has_text_modality=False)

    def test_config(self):
        self.config_tester.create_and_test_config_to_json_string()