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

[model] supported ERNIE4.5 Text Models (#9165)

Browse Source 
Co-authored-by: Yaowei Zheng <hiyouga@buaa.edu.cn>
This commit is contained in:
Hertz
2025-09-22 11:48:26 +08:00
committed by GitHub
parent 80fe3a172d
commit 344c760cc1
4 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -624,6 +624,25 @@ register_model_group(
)
register_model_group(
models={
"ERNIE-4.5-0.3B-PT": {
DownloadSource.DEFAULT: "baidu/ERNIE-4.5-0.3B-PT",
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-0.3B-PT",
},
"ERNIE-4.5-21B-A3B-PT": {
DownloadSource.DEFAULT: "baidu/ERNIE-4.5-21B-A3B-PT",
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-21B-A3B-PT",
},
"ERNIE-4.5-300B-A47B-PT": {
DownloadSource.DEFAULT: "baidu/ERNIE-4.5-300B-A47B-PT",
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-300B-A47B-PT",
},
},
template="ernie_nothink",
)
register_model_group(
models={
"EXAONE-3.0-7.8B-Instruct": {