diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py index 45145886..c413c51d 100644 --- a/src/llamafactory/extras/constants.py +++ b/src/llamafactory/extras/constants.py @@ -531,6 +531,14 @@ register_model_group( "Gemma-1.1-7B-Chat": { DownloadSource.DEFAULT: "google/gemma-1.1-7b-it", }, + "Gemma-2-2B": { + DownloadSource.DEFAULT: "google/gemma-2-2b", + DownloadSource.MODELSCOPE: "LLM-Research/gemma-2-2b", + }, + "Gemma-2-2B-Chat": { + DownloadSource.DEFAULT: "google/gemma-2-2b-it", + DownloadSource.MODELSCOPE: "LLM-Research/gemma-2-2b-it", + }, "Gemma-2-9B": { DownloadSource.DEFAULT: "google/gemma-2-9b", DownloadSource.MODELSCOPE: "LLM-Research/gemma-2-9b",