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

support Aya23

Browse Source 
Former-commit-id: e626e264460d12b282099bfbb8e6679c31e85fc0
This commit is contained in:
hiyouga 2024-05-27 20:23:24 +08:00
parent db569a2d61
commit fc5a6b5c4e
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": {