423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

support Aya23

Browse Source 
Former-commit-id: 071935b90006e2c79e39bb9ee0c5d48c6c910501
This commit is contained in:
hiyouga 2024-05-27 20:23:24 +08:00
parent 4bd276f58f
commit c43bc74fe6
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)
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(
models={
"Baichuan-7B-Base": {