Update path

Update preset path for keras.io. There is no LLaMA2 in keras.io https://keras.io/api/keras_hub/models/llama2 This is the actual link: https://keras.io/api/keras_hub/models/llama2 For Vicuna it does not have it's own model direcotry, since it is also the part of Llama,, updated the path.
keras-team · Oct 23, 2024 · 5967eec · 5967eec
1 parent 3b331dd
commit 5967eec
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/keras_hub/src/models/llama/llama_presets.py b/keras_hub/src/models/llama/llama_presets.py
@@ -7,7 +7,7 @@
             "description": "7 billion parameter, 32-layer, base LLaMA 2 model.",
             "params": 6738415616,
             "official_name": "LLaMA 2",
-            "path": "llama2",
+            "path": "llama",
             "model_card": "https://github.com/meta-llama/llama",
         },
         "kaggle_handle": "kaggle://keras/llama2/keras/llama2_7b_en/1",
@@ -20,7 +20,7 @@
             ),
             "params": 6739839488,
             "official_name": "LLaMA 2",
-            "path": "llama2",
+            "path": "llama",
             "model_card": "https://github.com/meta-llama/llama",
         },
         "kaggle_handle": "kaggle://keras/llama2/keras/llama2_7b_en_int8/1",
@@ -33,7 +33,7 @@
             ),
             "params": 6738415616,
             "official_name": "LLaMA 2",
-            "path": "llama2",
+            "path": "llama",
             "model_card": "https://github.com/meta-llama/llama",
         },
         "kaggle_handle": "kaggle://keras/llama2/keras/llama2_instruct_7b_en/1",
@@ -46,7 +46,7 @@
             ),
             "params": 6739839488,
             "official_name": "LLaMA 2",
-            "path": "llama2",
+            "path": "llama",
             "model_card": "https://github.com/meta-llama/llama",
         },
         "kaggle_handle": "kaggle://keras/llama2/keras/llama2_instruct_7b_en_int8/1",
@@ -59,7 +59,7 @@
             ),
             "params": 6738415616,
             "official_name": "Vicuna",
-            "path": "vicuna",
+            "path": "llama",
             "model_card": "https://github.com/lm-sys/FastChat",
         },
         "kaggle_handle": "kaggle://keras/vicuna/keras/vicuna_1.5_7b_en/1",