diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py index 1bd325c3..8a3ed036 100644 --- a/src/llamafactory/extras/constants.py +++ b/src/llamafactory/extras/constants.py @@ -1184,14 +1184,6 @@ register_model_group( DownloadSource.DEFAULT: "microsoft/Phi-3-mini-128k-instruct", DownloadSource.MODELSCOPE: "LLM-Research/Phi-3-mini-128k-instruct", }, - "Phi-3-7B-8k-Instruct": { - DownloadSource.DEFAULT: "microsoft/Phi-3-small-8k-instruct", - DownloadSource.MODELSCOPE: "LLM-Research/Phi-3-small-8k-instruct", - }, - "Phi-3-7B-128k-Instruct": { - DownloadSource.DEFAULT: "microsoft/Phi-3-small-128k-instruct", - DownloadSource.MODELSCOPE: "LLM-Research/Phi-3-small-128k-instruct", - }, "Phi-3-14B-8k-Instruct": { DownloadSource.DEFAULT: "microsoft/Phi-3-medium-4k-instruct", DownloadSource.MODELSCOPE: "LLM-Research/Phi-3-medium-4k-instruct", @@ -1205,6 +1197,21 @@ register_model_group( ) +register_model_group( + models={ + "Phi-3-7B-8k-Instruct": { + DownloadSource.DEFAULT: "microsoft/Phi-3-small-8k-instruct", + DownloadSource.MODELSCOPE: "LLM-Research/Phi-3-small-8k-instruct", + }, + "Phi-3-7B-128k-Instruct": { + DownloadSource.DEFAULT: "microsoft/Phi-3-small-128k-instruct", + DownloadSource.MODELSCOPE: "LLM-Research/Phi-3-small-128k-instruct", + }, + }, + template="phi_small", +) + + register_model_group( models={ "Pixtral-12B-Chat": {