diff --git a/paddleformers/transformers/download_utils.py b/paddleformers/transformers/download_utils.py index b6b83a078c7..eeae970ef18 100644 --- a/paddleformers/transformers/download_utils.py +++ b/paddleformers/transformers/download_utils.py @@ -711,57 +711,58 @@ } ) +# ernie register_model_group( models={ "ERNIE-4.5-300B-A47B-Base": { - DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-300B-A47B-Base-Paddle", - DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-300B-A47B-Base-Paddle", - DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-300B-A47B-Base-Paddle", + DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-300B-A47B-Base-PT", + DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-300B-A47B-Base-PT", + DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-300B-A47B-Base-PT", }, "ERNIE-4.5-300B-A47B": { - DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-300B-A47B-Paddle", - DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-300B-A47B-Paddle", - DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-300B-A47B-Paddle", + DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-300B-A47B-PT", + DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-300B-A47B-PT", + DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-300B-A47B-PT", }, "ERNIE-4.5-21B-A3B-Base": { - DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-21B-A3B-Base-Paddle", - DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-21B-A3B-Base-Paddle", - DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-21B-A3B-Base-Paddle", + DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-21B-A3B-Base-PT", + DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-21B-A3B-Base-PT", + DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-21B-A3B-Base-PT", }, "ERNIE-4.5-21B-A3B": { - DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-21B-A3B-Paddle", - DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-21B-A3B-Paddle", - DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-21B-A3B-Paddle", + DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-21B-A3B-PT", + DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-21B-A3B-PT", + DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-21B-A3B-PT", }, "ERNIE-4.5-0.3B-Base": { - DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-0.3B-Base-Paddle", - DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-0.3B-Base-Paddle", - DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-0.3B-Base-Paddle", + DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-0.3B-Base-PT", + DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-0.3B-Base-PT", + DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-0.3B-Base-PT", }, "ERNIE-4.5-0.3B": { - DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-0.3B-Paddle", - DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-0.3B-Paddle", - DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-0.3B-Paddle", + DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-0.3B-PT", + DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-0.3B-PT", + DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-0.3B-PT", }, "ERNIE-4.5-VL-424B-A47B-Base": { - DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-424B-A47B-Base-Paddle", - DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-424B-A47B-Base-Paddle", - DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-424B-A47B-Base-Paddle", + DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-424B-A47B-Base-PT", + DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-424B-A47B-Base-PT", + DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-424B-A47B-Base-PT", }, "ERNIE-4.5-VL-424B": { - DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-424B-Paddle", - DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-424B-Paddle", - DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-424B-Paddle", + DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-424B-PT", + DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-424B-PT", + DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-424B-PT", }, "ERNIE-4.5-VL-28B-A3B-Base": { - DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-28B-A3B-Base-Paddle", - DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Base-Paddle", - DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Base-Paddle", + DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-28B-A3B-Base-PT", + DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Base-PT", + DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Base-PT", }, "ERNIE-4.5-VL-28B-A3B": { - DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-28B-A3B-Paddle", - DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Paddle", - DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-Paddle", + DownloadSource.HUGGINGFACE: "baidu/ERNIE-4.5-VL-28B-A3B-PT", + DownloadSource.AISTUDIO: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-PT", + DownloadSource.MODELSCOPE: "PaddlePaddle/ERNIE-4.5-VL-28B-A3B-PT", }, } )