Skip to content

Commit 65a78bc

Browse files
committed
fix support ernie
1 parent 091fde8 commit 65a78bc

File tree

1 file changed

+30
-30
lines changed

1 file changed

+30
-30
lines changed

paddleformers/transformers/download_utils.py

Lines changed: 30 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -714,54 +714,54 @@
714714
register_model_group(
715715
models={
716716
"ERNIE-4.5-300B-A47B-Base": {
717-
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-300B-A47B-Base-Paddle",
718-
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-300B-A47B-Base-Paddle",
719-
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-300B-A47B-Base-Paddle",
717+
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-300B-A47B-Base-PT",
718+
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-300B-A47B-Base-PT",
719+
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-300B-A47B-Base-PT",
720720
},
721721
"ERNIE-4.5-300B-A47B": {
722-
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-300B-A47B-Paddle",
723-
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-300B-A47B-Paddle",
724-
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-300B-A47B-Paddle",
722+
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-300B-A47B-PT",
723+
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-300B-A47B-PT",
724+
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-300B-A47B-PT",
725725
},
726726
"ERNIE-4.5-21B-A3B-Base": {
727-
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-21B-A3B-Base-Paddle",
728-
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-21B-A3B-Base-Paddle",
729-
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-21B-A3B-Base-Paddle",
727+
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-21B-A3B-Base-PT",
728+
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-21B-A3B-Base-PT",
729+
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-21B-A3B-Base-PT",
730730
},
731731
"ERNIE-4.5-21B-A3B": {
732-
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-21B-A3B-Paddle",
733-
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-21B-A3B-Paddle",
734-
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-21B-A3B-Paddle",
732+
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-21B-A3B-PT",
733+
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-21B-A3B-PT",
734+
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-21B-A3B-PT",
735735
},
736736
"ERNIE-4.5-0.3B-Base": {
737-
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-0.3B-Base-Paddle",
738-
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-0.3B-Base-Paddle",
739-
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-0.3B-Base-Paddle",
737+
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-0.3B-Base-PT",
738+
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-0.3B-Base-PT",
739+
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-0.3B-Base-PT",
740740
},
741741
"ERNIE-4.5-0.3B": {
742-
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-0.3B-Paddle",
743-
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-0.3B-Paddle",
744-
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-0.3B-Paddle",
742+
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-0.3B-PT",
743+
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-0.3B-PT",
744+
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-0.3B-PT",
745745
},
746746
"ERNIE-4.5-VL-424B-A47B-Base": {
747-
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-424B-A47B-Base-Paddle",
748-
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-424B-A47B-Base-Paddle",
749-
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-424B-A47B-Base-Paddle",
747+
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-424B-A47B-Base-PT",
748+
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-424B-A47B-Base-PT",
749+
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-424B-A47B-Base-PT",
750750
},
751751
"ERNIE-4.5-VL-424B": {
752-
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-424B-Paddle",
753-
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-424B-Paddle",
754-
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-424B-Paddle",
752+
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-424B-PT",
753+
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-424B-PT",
754+
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-424B-PT",
755755
},
756756
"ERNIE-4.5-VL-28B-A3B-Base": {
757-
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-28B-A3B-Base-Paddle",
758-
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Base-Paddle",
759-
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Base-Paddle",
757+
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-28B-A3B-Base-PT",
758+
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Base-PT",
759+
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Base-PT",
760760
},
761761
"ERNIE-4.5-VL-28B-A3B": {
762-
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-28B-A3B-Paddle",
763-
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Paddle",
764-
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Paddle",
762+
DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-28B-A3B-PT",
763+
DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-PT",
764+
DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-PT",
765765
},
766766
}
767767
)

0 commit comments

Comments
 (0)