423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 11:50:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3254 from marko1616/feature/Add-support-for-CohereForAI/c4ai-command-r-plus

Browse Source 
Add template&support for c4ai-command-r/plus (tested)
This commit is contained in:
hoshi-hiyouga
2024-04-15 22:59:35 +08:00
committed by GitHub
parent cce52351b5 3ccf0d0977
commit 7a8ae3f4ac
2 changed files with 37 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/llmtuner/extras/constants.py
View File

@@ -242,6 +242,28 @@ register_model_group(
)
register_model_group(
models={
"CommandR-35B-Chat": {
DownloadSource.DEFAULT: "CohereForAI/c4ai-command-r-v01",
DownloadSource.MODELSCOPE: "AI-ModelScope/c4ai-command-r-v01",
},
"CommandR-Plus-104B-Chat": {
DownloadSource.DEFAULT: "CohereForAI/c4ai-command-r-plus",
DownloadSource.MODELSCOPE: "AI-ModelScope/c4ai-command-r-plus",
},
"CommandR-35B-4bit-Chat": {
DownloadSource.DEFAULT: "CohereForAI/c4ai-command-r-v01-4bit",
DownloadSource.MODELSCOPE: "mirror013/c4ai-command-r-v01-4bit",
},
"CommandR-Plus-104B-4bit-Chat": {
DownloadSource.DEFAULT: "CohereForAI/c4ai-command-r-plus-4bit",
},
},
template="cohere",
)
register_model_group(
models={
"DeepSeek-LLM-7B-Base": {