p04798526
/
LLaMA-Factory-Mirror
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

support Aya23

This commit is contained in:
hiyouga 2024-05-27 20:23:24 +08:00
parent 08bd0440b5
commit e626e26446
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -86,6 +86,19 @@ def register_model_group(
VISION_MODELS.add(prefix) VISION_MODELS.add(prefix)
register_model_group(
models={
"Aya-23-8B-Chat": {
DownloadSource.DEFAULT: "CohereForAI/aya-23-8B",
},
"Aya-23-35B-Chat": {
DownloadSource.DEFAULT: "CohereForAI/aya-23-35B",
},
},
template="cohere",
)
register_model_group( register_model_group(
models={ models={
"Baichuan-7B-Base": { "Baichuan-7B-Base": {