Fix all offload and MP tests (#17533)

This commit is contained in:
Sylvain Gugger
2022-06-03 09:59:13 -04:00
committed by GitHub
parent 1c57242d7b
commit 8343901263
4 changed files with 9 additions and 21 deletions

View File

@@ -63,7 +63,7 @@ class OPTModelTester:
use_labels=False,
vocab_size=99,
hidden_size=16,
num_hidden_layers=2,
num_hidden_layers=5,
num_attention_heads=4,
intermediate_size=4,
hidden_act="gelu",

View File

@@ -515,6 +515,8 @@ class T5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
test_resize_embeddings = True
test_model_parallel = True
is_encoder_decoder = True
# The small T5 model needs higher percentages for CPU/MP tests
model_split_percents = [0.8, 0.9]
def setUp(self):
self.model_tester = T5ModelTester(self)