SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[LongT5] disable model parallel test (#17702)

Browse Source
This commit is contained in:
Suraj Patil
2022-06-14 17:27:39 +02:00
committed by GitHub
parent 7ec9128e5a
commit 120649bf3a
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/longt5/test_modeling_longt5.py
View File

@@ -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):
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)