Fix test_model_parallelism (#25359)

* fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-08-08 10:48:45 +02:00
parent d4bd33cc9f
commit 6ea3ee3cd2
25 changed files with 37 additions and 15 deletions
--- a/tests/models/roberta_prelayernorm/test_modeling_roberta_prelayernorm.py
+++ b/tests/models/roberta_prelayernorm/test_modeling_roberta_prelayernorm.py
@@ -395,6 +395,7 @@ class RobertaPreLayerNormModelTest(ModelTesterMixin, GenerationTesterMixin, Pipe
        else {}
    )
    fx_compatible = False
+    model_split_percents = [0.5, 0.8, 0.9]

    def setUp(self):
        self.model_tester = RobertaPreLayerNormModelTester(self)