diff --git a/tests/test_tokenizer.py b/tests/test_tokenizer.py index f9aede3921..b7e4d0b33f 100644 --- a/tests/test_tokenizer.py +++ b/tests/test_tokenizer.py @@ -11,6 +11,7 @@ from litgpt.tokenizer import Tokenizer +@pytest.mark.flaky(reruns=5) @pytest.mark.parametrize("config", config_module.configs, ids=[c["hf_config"]["name"] for c in config_module.configs]) def test_tokenizer_against_hf(config): access_token = os.getenv("HF_TOKEN")