fix accelerate tests for roberta xl (#31288)

* fix accelerate tests for roberta xl * style
2024-06-06 15:44:35 +02:00
parent 5ba8ac54f5
commit 99895ae5e2
2 changed files with 3 additions and 1 deletions
--- a/tests/models/xlm_roberta_xl/test_modeling_xlm_roberta_xl.py
+++ b/tests/models/xlm_roberta_xl/test_modeling_xlm_roberta_xl.py
@@ -387,6 +387,8 @@ class XLMRobertaXLModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTes
        else {}
    )

+    model_split_percents = [0.5, 0.85, 0.95]
+
    # TODO: Fix the failed tests
    def is_pipeline_test_to_skip(
        self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name