From 5967eec88bdb718c9e0eb4eb87c3f5f390c8f243 Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Wed, 23 Oct 2024 15:39:12 -0700 Subject: [PATCH] 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_hub/src/models/llama/llama_presets.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/keras_hub/src/models/llama/llama_presets.py b/keras_hub/src/models/llama/llama_presets.py index 6197cfe07f..f72a0ec95f 100644 --- 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",