add om_hub_token argument

Former-commit-id: b3214e69d32067a1c22dbd60c2cde1545ba75b19
2024-10-10 17:16:46 +08:00
parent aad79127e6
commit 5b15ca0b0b
3 changed files with 6 additions and 2 deletions
--- a/src/llamafactory/extras/constants.py
+++ b/src/llamafactory/extras/constants.py
@@ -563,7 +563,7 @@ register_model_group(
        "Gemma-2-9B-Instruct": {
            DownloadSource.DEFAULT: "google/gemma-2-9b-it",
            DownloadSource.MODELSCOPE: "LLM-Research/gemma-2-9b-it",
-            DownloadSource.MODELERS: "LlamaFactory/Qwen2-VL-2B-Instruct"
+            DownloadSource.MODELERS: "LlamaFactory/gemma-2-2b-it"
        },
        "Gemma-2-27B-Instruct": {
            DownloadSource.DEFAULT: "google/gemma-2-27b-it",