"test/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "45bb91ea71622d2b5b1a650df857ee2262c98668"
Unverified Commit e30bbb26 authored by Joao Gante's avatar Joao Gante Committed by GitHub
Browse files

Tests: tag `test_save_load_fast_init_from_base` as flaky (#28930)

parent 1709886e
......@@ -64,6 +64,7 @@ from transformers.models.auto.modeling_auto import (
)
from transformers.testing_utils import (
CaptureLogger,
is_flaky,
is_pt_flax_cross_test,
is_pt_tf_cross_test,
require_accelerate,
......@@ -381,6 +382,7 @@ class ModelTesterMixin:
m.gradient_checkpointing, f"Module {n} does not have gradient_checkpointing set to False"
)
@is_flaky(description="low likelihood of failure, reason not yet discovered")
def test_save_load_fast_init_from_base(self):
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
if config.__class__ not in MODEL_MAPPING:
......
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