Update VitDetModelTester.get_config to use pretrain_image_size (h…

…uggingface#27831) fix Co-authored-by: ydshieh <[email protected]>
namctin · Dec 5, 2023 · ac97507 · ac97507
1 parent 28e2887
commit ac97507
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/models/vitdet/test_modeling_vitdet.py b/tests/models/vitdet/test_modeling_vitdet.py
@@ -90,6 +90,7 @@ def prepare_config_and_inputs(self):
     def get_config(self):
         return VitDetConfig(
             image_size=self.image_size,
+            pretrain_image_size=self.image_size,
             patch_size=self.patch_size,
             num_channels=self.num_channels,
             hidden_size=self.hidden_size,