Skip test_multi_gpu_data_parallel_forward for UperNetModelTest (h…

…uggingface#21216)

fix

Co-authored-by: ydshieh <[email protected]>

tests/models/upernet/test_modeling_upernet.py

@@ -20,7 +20,7 @@
 
 from huggingface_hub import hf_hub_download
 from transformers import ConvNextConfig, UperNetConfig
-from transformers.testing_utils import require_torch, require_vision, slow, torch_device
+from transformers.testing_utils import require_torch, require_torch_multi_gpu, require_vision, slow, torch_device
 from transformers.utils import is_torch_available, is_vision_available
 
 from ...test_configuration_common import ConfigTester
@@ -199,6 +199,11 @@ def test_save_load_fast_init_from_base(self):
     def test_save_load_fast_init_to_base(self):
         pass
 
+    @require_torch_multi_gpu
+    @unittest.skip(reason="UperNet has some layers using `add_module` which doesn't work well with `nn.DataParallel`")
+    def test_multi_gpu_data_parallel_forward(self):
+        pass
+
     def test_hidden_states_output(self):
         def check_hidden_states_output(inputs_dict, config, model_class):
             model = model_class(config)