diff --git a/tests/models/align/test_modeling_align.py b/tests/models/align/test_modeling_align.py index 02d046924..2357c20e2 100644 --- a/tests/models/align/test_modeling_align.py +++ b/tests/models/align/test_modeling_align.py @@ -437,6 +437,10 @@ class AlignModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): config_and_inputs = self.model_tester.prepare_config_and_inputs() self.model_tester.create_and_check_model(*config_and_inputs) + @unittest.skip(reason="Start to fail after using torch `cu118`.") + def test_multi_gpu_data_parallel_forward(self): + super().test_multi_gpu_data_parallel_forward() + @unittest.skip(reason="Hidden_states is tested in individual model tests") def test_hidden_states_output(self): pass