"scripts/vscode:/vscode.git/clone" did not exist on "c7bfdb1cca2c411fc9e9c8ca87847f52c29ba31c"
Unverified Commit 99895ae5 authored by Marc Sun's avatar Marc Sun Committed by GitHub
Browse files

fix accelerate tests for roberta xl (#31288)

* fix accelerate tests for roberta xl

* style
parent 5ba8ac54
...@@ -572,7 +572,7 @@ class XLMRobertaXLPreTrainedModel(PreTrainedModel): ...@@ -572,7 +572,7 @@ class XLMRobertaXLPreTrainedModel(PreTrainedModel):
config_class = XLMRobertaXLConfig config_class = XLMRobertaXLConfig
base_model_prefix = "roberta" base_model_prefix = "roberta"
_no_split_modules = ["XLMRobertaXLEmbeddings", "XLMRobertaXLSelfAttention"] _no_split_modules = ["XLMRobertaXLEmbeddings", "XLMRobertaXLLayer"]
# Copied from transformers.models.bert.modeling_bert.BertPreTrainedModel._init_weights # Copied from transformers.models.bert.modeling_bert.BertPreTrainedModel._init_weights
def _init_weights(self, module): def _init_weights(self, module):
......
...@@ -387,6 +387,8 @@ class XLMRobertaXLModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTes ...@@ -387,6 +387,8 @@ class XLMRobertaXLModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTes
else {} else {}
) )
model_split_percents = [0.5, 0.85, 0.95]
# TODO: Fix the failed tests # TODO: Fix the failed tests
def is_pipeline_test_to_skip( def is_pipeline_test_to_skip(
self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name self, pipeline_test_casse_name, config_class, model_architecture, tokenizer_name, processor_name
......
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