Fix a BridgeTower test (huggingface#23694)

fix Co-authored-by: ydshieh <[email protected]>
novice03 · Jun 23, 2023 · 4671310 · 4671310
1 parent 2c10974
commit 4671310
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/models/bridgetower/test_modeling_bridgetower.py b/tests/models/bridgetower/test_modeling_bridgetower.py
@@ -627,7 +627,8 @@ def _get_non_used_layer_names(self, model_class):
         non_used_layer_names = ["text_model.pooler"]
         if model_class == BridgeTowerForMaskedLM:
             non_used_layer_names = non_used_layer_names + [
-                "cross_modal_image_layers.5",
+                # This number `1` actually depends on the number of layers in `cross_modal_image_layers` (by minus 1)
+                "cross_modal_image_layers.1",
                 "cross_modal_image_pooler",
                 "cross_modal_text_pooler",
             ]