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_load_model.py
+++ b/tests/e2e/test_load_model.py
@@ -26,9 +26,8 @@ class TestLoadModelUtils:
        # load config
        self.cfg = DictDefault(
            {
-                "base_model": "JackFram/llama-68m",
-                "tokenizer_type": "LlamaTokenizer",
-                "tokenizer_config": "JackFram/llama-68m",
+                "base_model": "HuggingFaceTB/SmolLM2-135M-Instruct",
+                "tokenizer_config": "HuggingFaceTB/SmolLM2-135M-Instruct",
                "sequence_len": 1024,
                "load_in_8bit": False,
                "adapter": "lora",