Skip to content

GitLab

Unverified Commit 120649bf authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

[LongT5] disable model parallel test (#17702)

parent 7ec9128e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
tests/models/longt5/test_modeling_longt5.py
View file @ 120649bf
...@@ -505,11 +505,10 @@ class LongT5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase ...@@ -505,11 +505,10 @@ class LongT5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase
all_model_classes = (LongT5Model, LongT5ForConditionalGeneration) if is_torch_available() else () all_model_classes = (LongT5Model, LongT5ForConditionalGeneration) if is_torch_available() else ()
all_generative_model_classes = (LongT5ForConditionalGeneration,) if is_torch_available() else () all_generative_model_classes = (LongT5ForConditionalGeneration,) if is_torch_available() else ()
fx_compatible = False fx_compatible = False
all_parallelizable_model_classes = (LongT5Model, LongT5ForConditionalGeneration) if is_torch_available() else ()
test_pruning = False test_pruning = False
test_torchscript = True test_torchscript = True
test_resize_embeddings = True test_resize_embeddings = True
test_model_parallel = True test_model_parallel = False
is_encoder_decoder = True is_encoder_decoder = True
def setUp(self): def setUp(self):
...@@ -1013,8 +1012,7 @@ class LongT5EncoderOnlyModelTest(ModelTesterMixin, unittest.TestCase): ...@@ -1013,8 +1012,7 @@ class LongT5EncoderOnlyModelTest(ModelTesterMixin, unittest.TestCase):
test_pruning = False test_pruning = False
test_torchscript = True test_torchscript = True
test_resize_embeddings = False test_resize_embeddings = False
test_model_parallel = True test_model_parallel = False
all_parallelizable_model_classes = (LongT5EncoderModel,) if is_torch_available() else ()
def setUp(self): def setUp(self):
self.model_tester = LongT5EncoderOnlyModelTester(self) self.model_tester = LongT5EncoderOnlyModelTester(self)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment