swap tinymodels that have safetensors for some ci tests (#2641)

2025-05-07 15:06:07 -04:00
parent 25e6c5f9bd
commit 0f3587174d
14 changed files with 137 additions and 20 deletions
--- a/tests/e2e/multigpu/test_llama.py
+++ b/tests/e2e/multigpu/test_llama.py
@@ -479,7 +479,7 @@ class TestMultiGPULlama:
                "sample_packing": True,
                "pad_to_sequence_len": True,
                "sequence_len": 2048,
-                "val_set_size": 0.05,
+                "val_set_size": 0.1,
                "special_tokens": {
                    "pad_token": "<|endoftext|>",
                },