From 120649bf3ad212fe25849b14d766a255bda3ea80 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Tue, 14 Jun 2022 17:27:39 +0200 Subject: [PATCH] [LongT5] disable model parallel test (#17702) --- tests/models/longt5/test_modeling_longt5.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/models/longt5/test_modeling_longt5.py b/tests/models/longt5/test_modeling_longt5.py index d0bf1ac804..65375e0faf 100644 --- a/tests/models/longt5/test_modeling_longt5.py +++ b/tests/models/longt5/test_modeling_longt5.py @@ -505,11 +505,10 @@ class LongT5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase all_model_classes = (LongT5Model, LongT5ForConditionalGeneration) if is_torch_available() else () all_generative_model_classes = (LongT5ForConditionalGeneration,) if is_torch_available() else () fx_compatible = False - all_parallelizable_model_classes = (LongT5Model, LongT5ForConditionalGeneration) if is_torch_available() else () test_pruning = False test_torchscript = True test_resize_embeddings = True - test_model_parallel = True + test_model_parallel = False is_encoder_decoder = True def setUp(self): @@ -1013,8 +1012,7 @@ class LongT5EncoderOnlyModelTest(ModelTesterMixin, unittest.TestCase): test_pruning = False test_torchscript = True test_resize_embeddings = False - test_model_parallel = True - all_parallelizable_model_classes = (LongT5EncoderModel,) if is_torch_available() else () + test_model_parallel = False def setUp(self): self.model_tester = LongT5EncoderOnlyModelTester(self)