From 5e005262b4f55d0e2415291292b02ffa67488142 Mon Sep 17 00:00:00 2001 From: Divyashree Sreepathihalli Date: Tue, 21 May 2024 10:56:47 -0700 Subject: [PATCH] Update pali_gemma_presets.py --- keras_nlp/src/models/pali_gemma/pali_gemma_presets.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/keras_nlp/src/models/pali_gemma/pali_gemma_presets.py b/keras_nlp/src/models/pali_gemma/pali_gemma_presets.py index 8cabcc7038..c8791f709b 100644 --- a/keras_nlp/src/models/pali_gemma/pali_gemma_presets.py +++ b/keras_nlp/src/models/pali_gemma/pali_gemma_presets.py @@ -25,7 +25,7 @@ "path": "pali_gemma", "model_card": "https://www.kaggle.com/models/google/paligemma", }, - "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_mix_224/2", + "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_mix_224/1", }, "pali_gemma_3b_mix_448": { "metadata": { @@ -37,7 +37,7 @@ "path": "pali_gemma", "model_card": "https://www.kaggle.com/models/google/paligemma", }, - "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_mix_448/2", + "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_mix_448/1", }, "pali_gemma_3b_224": { "metadata": { @@ -49,7 +49,7 @@ "path": "pali_gemma", "model_card": "https://www.kaggle.com/models/google/paligemma", }, - "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_224/2", + "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_224/1", }, "pali_gemma_3b_448": { "metadata": { @@ -61,7 +61,7 @@ "path": "pali_gemma", "model_card": "https://www.kaggle.com/models/google/paligemma", }, - "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_448/2", + "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_448/1", }, "pali_gemma_3b_896": { "metadata": { @@ -73,6 +73,6 @@ "path": "pali_gemma", "model_card": "https://www.kaggle.com/models/google/paligemma", }, - "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_896/2", + "kaggle_handle": "kaggle://keras/paligemma/keras/pali_gemma_3b_896/1", }, }