fix preset name

keras-team · Oct 17, 2024 · 2cdbfc2 · 2cdbfc2
1 parent 58326f8
commit 2cdbfc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/checkpoint_conversion/convert_pali_gemma_checkpoints.py b/tools/checkpoint_conversion/convert_pali_gemma_checkpoints.py
@@ -4,7 +4,7 @@
     --image_size=224 --checkpoint_name=pali_gemma_3b_mix_224
 python tools/checkpoint_conversion/convert_pali_gemma_checkpoints.py \
   --weights_path=paligemma-3b-mix-448.npz \
-    --image_size=448 --checkpoint_name=pali_gemma_3b_mix_428
+    --image_size=448 --checkpoint_name=pali_gemma_3b_mix_448
 python tools/checkpoint_conversion/convert_pali_gemma_checkpoints.py \
   --weights_path=paligemma-3b-pt-224.npz \
     --image_size=224 --checkpoint_name=pali_gemma_3b_224