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/test_normalize_config.py
+++ b/tests/test_normalize_config.py
@@ -16,9 +16,8 @@ class NormalizeConfigTestCase(unittest.TestCase):
    def _get_base_cfg(self):
        return DictDefault(
            {
-                "base_model": "JackFram/llama-68m",
-                "base_model_config": "JackFram/llama-68m",
-                "tokenizer_type": "LlamaTokenizer",
+                "base_model": "HuggingFaceTB/SmolLM2-135M-Instruct",
+                "base_model_config": "HuggingFaceTB/SmolLM2-135M-Instruct",
                "num_epochs": 1,
                "micro_batch_size": 1,
                "gradient_accumulation_steps": 1,