423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-20 12:42:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

[misc] fix gemma template (#8970)

Browse Source
This commit is contained in:
Yaowei Zheng 2025-08-20 03:18:18 +08:00 committed by GitHub
parent 44beb76cdf
commit ac0f50a09a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/llamafactory/extras/constants.py
View File

@ -757,10 +757,18 @@ register_model_group(
DownloadSource.DEFAULT: "google/gemma-2-27b-it", DownloadSource.DEFAULT: "google/gemma-2-27b-it",
DownloadSource.MODELSCOPE: "LLM-Research/gemma-2-27b-it", DownloadSource.MODELSCOPE: "LLM-Research/gemma-2-27b-it",
}, },
"Gemma-3-270M": {
DownloadSource.DEFAULT: "google/gemma-3-270m",
DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-270m",
},
"Gemma-3-1B": { "Gemma-3-1B": {
DownloadSource.DEFAULT: "google/gemma-3-1b-pt", DownloadSource.DEFAULT: "google/gemma-3-1b-pt",
DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-1b-pt", DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-1b-pt",
}, },
"Gemma-3-270M-Instruct": {
DownloadSource.DEFAULT: "google/gemma-3-270m-it",
DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-270m-it",
},
"Gemma-3-1B-Instruct": { "Gemma-3-1B-Instruct": {
DownloadSource.DEFAULT: "google/gemma-3-1b-it", DownloadSource.DEFAULT: "google/gemma-3-1b-it",
DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-1b-it", DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-1b-it",
@ -776,10 +784,6 @@ register_model_group(
register_model_group( register_model_group(
models={ models={
"Gemma-3-270M": {
DownloadSource.DEFAULT: "google/gemma-3-270m",
DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-270m",
},
"Gemma-3-4B": { "Gemma-3-4B": {
DownloadSource.DEFAULT: "google/gemma-3-4b-pt", DownloadSource.DEFAULT: "google/gemma-3-4b-pt",
DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-4b-pt", DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-4b-pt",
@ -792,10 +796,6 @@ register_model_group(
DownloadSource.DEFAULT: "google/gemma-3-27b-pt", DownloadSource.DEFAULT: "google/gemma-3-27b-pt",
DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-27b-pt", DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-27b-pt",
}, },
"Gemma-3-270M-Instruct": {
DownloadSource.DEFAULT: "google/gemma-3-270m-it",
DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-270m-it",
},
"Gemma-3-4B-Instruct": { "Gemma-3-4B-Instruct": {
DownloadSource.DEFAULT: "google/gemma-3-4b-it", DownloadSource.DEFAULT: "google/gemma-3-4b-it",
DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-4b-it", DownloadSource.MODELSCOPE: "LLM-Research/gemma-3-4b-it",
@ -816,6 +816,10 @@ register_model_group(
DownloadSource.DEFAULT: "google/medgemma-4b-it", DownloadSource.DEFAULT: "google/medgemma-4b-it",
DownloadSource.MODELSCOPE: "google/medgemma-4b-it", DownloadSource.MODELSCOPE: "google/medgemma-4b-it",
}, },
"MedGemma-27B-Instruct": {
DownloadSource.DEFAULT: "google/medgemma-27b-text-it",
DownloadSource.MODELSCOPE: "google/medgemma-27b-text-it",
},
}, },
template="gemma3", template="gemma3",
multimodal=True, multimodal=True,