fix(test): replace jackfram llama with smollm

2025-02-28 16:40:49 +07:00
parent 75cbd15301
commit 10d18e6c97
12 changed files with 26 additions and 41 deletions
--- a/tests/e2e/test_llama.py
+++ b/tests/e2e/test_llama.py
@@ -26,8 +26,7 @@ class TestLlama:
        # pylint: disable=duplicate-code
        cfg = DictDefault(
            {
-                "base_model": "JackFram/llama-68m",
-                "tokenizer_type": "LlamaTokenizer",
+                "base_model": "HuggingFaceTB/SmolLM2-135M-Instruct",
                "trust_remote_code": True,
                "sequence_len": 512,
                "val_set_size": 0.1,