Added tinyllama downloading

krai · Oct 24, 2024 · 82d4ec6 · 82d4ec6
1 parent c08f9f8
commit 82d4ec6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/llm_hf_weights_recipe/data_axs.json b/llm_hf_weights_recipe/data_axs.json
@@ -14,6 +14,16 @@
     "variant": "7b",
 
     "model_family_information": {
+        "tiniest_llama": {
+            "name_format": "tinyllama-#{variant}#",
+            "model_publisher": "nickypro",
+            "allowed_variants": ["15m"]
+        },
+        "tiny_llama": {
+            "name_format": "TinyLlama-#{variant}#-Chat-v1.0",
+            "model_publisher": "TinyLlama",
+            "allowed_variants": ["1.1b"]
+        },
         "llama2": {
             "name_format": "Llama-2-#{variant}#-chat-hf",
             "model_publisher": "meta-llama",