资讯 文档
技术能力
语音技术
文字识别
人脸与人体
图像技术
语言与知识
视频技术

模型的下载

引言

飞桨AI Studio模型库提供了丰富的预训练模型资源,涵盖计算机视觉、自然语言处理、语音识别等多个领域。本文档将详细介绍以下四种高效的模型下载方式,帮助您快速获取所需模型资源。

  • 使用命令行下载
  • 使用 SDK 下载
  • 使用 Git 下载
  • 网页下载

1. 命令行下载

aistudio download -h                               
usage: PROG download [-h] [--model MODEL] [--revision REVISION] [--local_dir LOCAL_DIR] [--include [INCLUDE ...]] [--exclude [EXCLUDE ...]]
                     [--max-workers MAX_WORKERS]
                     [files ...]

positional arguments:
  files                 Specify relative path to the repository file(s) to download.(e.g 'tokenizer.json', 'dir/decoder_model.onnx').

options:
  -h, --help            show this help message and exit
  --model MODEL         模型ID,例如 myname/myrepoid
  --revision REVISION   Revision of the entity.
  --local_dir LOCAL_DIR
                        File will be downloaded to local location specified bylocal_dir, in this case.
  --include [INCLUDE ...]
                        Glob patterns to match files to download.Ignored if file is specified
  --exclude [EXCLUDE ...]
                        Glob patterns to exclude from files to download.Ignored if file is specified
  --max-workers MAX_WORKERS
                        The maximum number of workers to download files.

1.1 下载整个repo(到默认cache地址)

# 在{repo_id}处填写需要下载的模型repo_id,如PaddlePaddle/ERNIE4.5
aistudio download --model {repo_id}

1.2 下载整个repo(到指定目录)

# 在{repo_id}处填写需要下载的模型repo_id,在local_dir后指定下载路径,以下示例为下载到当前文件夹
aistudio download --model {repo_id} --local_dir ./

1.3 指定下载单个文件(到指定目录)

# 以下载README.md为例
aistudio download files README.md --model {repo_id} --local_dir ./

1.4 指定下载单个文件夹(到指定目录)

# 以下载文件夹名称为“folder_name”为例
aistudio download --model {repo_id} --include '{folder_name}/*' --local_dir ./

2. SDK下载(Python环境)

2.1 环境准备

# 首先请先安装aistudio-sdk库
pip install --upgrade aistudio-sdk

2.2 下载整个仓库

import os
# 如下载私密模型,需要填写aistudio-access-token, 位置在我的控制台--令牌
os.environ["AISTUDIO_ACCESS_TOKEN"] = "INPUT_YOUR_AISTUDIO_TOKEN"
from aistudio_sdk.snapshot_download import snapshot_download

res = snapshot_download(
    # 填写模型详情页面中的repo_id,如myname/myrepoid
    repo_id='myname/myrepoid',
    # 填写分支版本,如master
    revision='master',
    # 填写本地保存路径,如./
    local_dir='./'
)
print(res)

参数说明(snapshot_download):

字段名 必填 类型 描述
repo_id str 模型唯一标识符(如 PaddlePaddle/ERNIE4.5
revision str 模型的Git版本(分支/tag),默认 master
allow_patterns str,List 指定要下载的文件模式,如文件名或文件扩展名
ignore_patterns str,List 指定要忽略下载的文件模式,如文件名或文件扩展名
local_dir str 指定模型的下载存放目录,给定后本次下载的模型文件将会被存储在 local_dir/模型文件

2.3 下载指定文件

import os
# 如下载私密模型,需要填写aistudio-access-token, 位置在我的控制台--令牌
os.environ["AISTUDIO_ACCESS_TOKEN"] = "INPUT_YOUR_AISTUDIO_TOKEN"
from aistudio_sdk.file_download import model_file_download

res = model_file_download(
    # 填写模型详情页面中的repo_id,如myname/myrepoid
    repo_id='myname/myrepoid',
    # 填写需要下载的文件路径,如mydir/myfile.txt
    file_path='README.md',
    # 填写分支版本,如master
    revision='master',
    # 填写本地保存路径,如./
    local_dir='./'
)
print(res)

参数说明(file_download):

字段名 必填 类型 描述
repo_id str 模型唯一标识符(如 PaddlePaddle/ERNIE4.5
file_path str 待下载文件在模型仓库的相对路径,不支持文件夹。如 inference_model/model.pdparams
revision str 模型的Git版本(分支/tag),默认 master
local_dir str 指定模型的下载存放目录,给定后本次下载的模型文件将会被存储在 local_dir/模型文件

2.4 下载指定文件夹

import os
# 如下载私密模型,需要填写aistudio-access-token, 位置在我的控制台--令牌
os.environ["AISTUDIO_ACCESS_TOKEN"] = "INPUT_YOUR_AISTUDIO_TOKEN"
from aistudio_sdk.snapshot_download import snapshot_download

res = snapshot_download(
    # 填写模型详情页面中的repo_id,如myname/myrepoid
    repo_id='myname/myrepoid',
    # 填写分支版本,如master
    revision='master',
    # 填写本地保存路径,如./
    local_dir='./'
    # 填写所需下载的文件夹名称,如folder_name/*
    allow_patterns='folder_name/*'
)
print(res)

参数说明(snapshot_download):

字段名 必填 类型 描述
repo_id str 模型唯一标识符(如 PaddlePaddle/ERNIE4.5
revision str 模型的Git版本(分支/tag),默认 master
allow_patterns str,List 指定要下载的文件模式,如文件名或文件扩展名
ignore_patterns str,List 指定要忽略下载的文件模式,如文件名或文件扩展名
local_dir str 指定模型的下载存放目录,给定后本次下载的模型文件将会被存储在 local_dir/模型文件

3. Git仓库下载

下载公开模型

# 请确认已经安装git-lfs (https://git-lfs.com)
git lfs install
# 在{repo_id}处填写需要下载的模型repo_id,如PaddlePaddle/ERNIE4.5
# 示例:git clone https://git.aistudio.baidu.com/paddlepaddle/ernie4.5.git
git clone https://git.aistudio.baidu.com/{repo_id}.git

下载私密模型

# 请确认已经安装git-lfs (https://git-lfs.com)
git lfs install
# 在{token}处填写您的访问令牌,可以在我的工作台-令牌处查看,在{repo_id}处填写需要下载的模型repo_id
# 示例:git clone https://6b27a165b0851ae94b4651fbc5fe0902e2cd17bc@git.aistudio.baidu.com/paddlepaddle/ernie4.5.git
git clone https://{token}@git.aistudio.baidu.com/{repo_id}.git

跳过大文件(LFS)下载

如果需要跳过LFS大文件的下载,可以在git clone命令前添加GIT_LFS_SKIP_SMUDGE=1

GIT_LFS_SKIP_SMUDGE=1 git clone https://{token}@git.aistudio.baidu.com/{repo_id}.git

4. 网页下载

操作步骤

  1. 访问飞桨AI Studio模型库
  2. 点击进入到目标模型详情页
  3. 在"模型空间"中浏览模型文件
  4. 点击文件右侧的下载按钮


上一篇
模型创建与编辑
下一篇
模型的上传