diff --git a/src/llamafactory/extras/constants.py b/src/llamafactory/extras/constants.py index 4d7685c5..687e16cc 100644 --- a/src/llamafactory/extras/constants.py +++ b/src/llamafactory/extras/constants.py @@ -301,6 +301,16 @@ register_model_group( ) +register_model_group( + models={ + "Codestral-22B-v0.1-Chat": { + DownloadSource.DEFAULT: "mistralai/Codestral-22B-v0.1", + }, + }, + template="mistral", +) + + register_model_group( models={ "CommandR-35B-Chat": {