Merge branch 'main' into litgpt-serve

Lightning-AI · Apr 16, 2024 · 75f017e · 75f017e
2 parents 7aaa778 + 1245b26
commit 75f017e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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")