Improve the documents for PaliGemma2

ufownl · Dec 5, 2024 · abc9a2e · abc9a2e
1 parent 4f8cbb3
commit abc9a2e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -188,6 +188,8 @@ Available options:
                            -- 27b-it (Gemma2 27B parameters, instruction-tuned),
                            -- 27b-pt (Gemma2 27B parameters, pretrained),
                            -- paligemma-224 (PaliGemma 224*224),
+                           -- paligemma2-3b-224 (PaliGemma2 3B 224*224),
+                           -- paligemma2-10b-224 (PaliGemma2 10B 224*224),
                            -- (required)
   weights = "/path/to/2.0-2b-it-sfp.sbs",  -- Path of model weights file. (required)
   weight_type = "sfp",  -- Weight type:

diff --git a/examples/argparse.lua b/examples/argparse.lua
@@ -31,6 +31,8 @@ function _M.help(description, usage)
   print("    9b-it = Gemma2 9B parameters, instruction-tuned")
   print("    27b-it = Gemma2 27B parameters, instruction-tuned")
   print("    paligemma-224 = PaliGemma 224*224")
+  print("    paligemma2-3b-224 = PaliGemma2 3B 224*224")
+  print("    paligemma2-10b-224 = PaliGemma2 10B 224*224")
   print("  --weights: Path of model weights file. (default: 2.0-2b-it-sfp.sbs)")
   print("  --weight_type: Weight type (default: sfp)")
 end