Fix tests

huggingface · Feb 19, 2024 · 4d524f2 · 4d524f2
1 parent 0a38d16
commit 4d524f2
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/src/transformers/models/dpt/modeling_dpt.py b/src/transformers/models/dpt/modeling_dpt.py
@@ -1079,7 +1079,7 @@ def __init__(self, config):
         super().__init__(config)
 
         self.backbone = None
-        if config.is_hybrid:
+        if config.is_hybrid or config.backbone_config is None:
             self.dpt = DPTModel(config, add_pooling_layer=False)
         else:
             self.backbone = load_backbone(config)

diff --git a/tests/models/conditional_detr/test_modeling_conditional_detr.py b/tests/models/conditional_detr/test_modeling_conditional_detr.py
@@ -443,7 +443,9 @@ def test_different_timm_backbone(self):
 
         # let's pick a random timm backbone
         config.backbone = "tf_mobilenetv3_small_075"
+        config.backbone_config = None
         config.use_timm_backbone = True
+        config.backbone_kwargs = {"out_indices": [2, 3, 4]}
 
         for model_class in self.all_model_classes:
             model = model_class(config)

diff --git a/tests/models/detr/test_modeling_detr.py b/tests/models/detr/test_modeling_detr.py
@@ -473,7 +473,6 @@ def test_greyscale_images(self):
         )
 
         # let's set num_channels to 1
-        config.num_channels = 1
         config.backbone_config.num_channels = 1
 
         for model_class in self.all_model_classes:

diff --git a/tests/models/table_transformer/test_modeling_table_transformer.py b/tests/models/table_transformer/test_modeling_table_transformer.py
@@ -455,6 +455,9 @@ def test_different_timm_backbone(self):
 
         # let's pick a random timm backbone
         config.backbone = "tf_mobilenetv3_small_075"
+        config.backbone_config = None
+        config.use_timm_backbone = True
+        config.backbone_kwargs = {"out_indices": [2, 3, 4]}
 
         for model_class in self.all_model_classes:
             model = model_class(config)
@@ -482,7 +485,6 @@ def test_greyscale_images(self):
         )
 
         # let's set num_channels to 1
-        config.num_channels = 1
         config.backbone_config.num_channels = 1
 
         for model_class in self.all_model_classes: