kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add falcon 11b

Browse Source
This commit is contained in:
hiyouga committed 2024-05-17 00:08:33 +08:00
1 parent 3df986c679
commit d77bed4091
3 files changed
+5 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/llamafactory/extras/constants.py
+3
View File
@@ -375,6 +375,9 @@ register_model_group(
DownloadSource.DEFAULT: "tiiuae/falcon-7b",
DownloadSource.MODELSCOPE: "AI-ModelScope/falcon-7b",
},
"Falcon-11B": {
DownloadSource.DEFAULT: "tiiuae/falcon-11B",
},
"Falcon-40B": {
DownloadSource.DEFAULT: "tiiuae/falcon-40b",
DownloadSource.MODELSCOPE: "AI-ModelScope/falcon-40b",