From bde7378bf0fad2ce769ae2a5f9e5c055f076bcb5 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 24 Jan 2023 10:41:16 +0100 Subject: [PATCH] Skip `test_multi_gpu_data_parallel_forward` for `UperNetModelTest` (#21216) fix Co-authored-by: ydshieh --- tests/models/upernet/test_modeling_upernet.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/models/upernet/test_modeling_upernet.py b/tests/models/upernet/test_modeling_upernet.py index 8845edd2d4..f3982a79df 100644 --- a/tests/models/upernet/test_modeling_upernet.py +++ b/tests/models/upernet/test_modeling_upernet.py @@ -20,7 +20,7 @@ import unittest 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 @@ class UperNetModelTest(ModelTesterMixin, unittest.TestCase): 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)