跳到主要内容

模型下载

我们提供两种方式下载模型：克隆仓库和使用 huggingface_hub transfomers diffusers 库直接使用。

一. 克隆仓库​

请参考 模型仓库 文档，做好必要设置后，即可克隆仓库到本地。

二. 使用 huggingface 库下载模型​

我们对 huggingface 模型下载相关 API 进行了适配，只需要配置合适的环境变量 HF_ENDPOINT 和 HF_HOME，即可使用 huggingface_hub transfomers diffusers 库直接下载使用模力方舟上的模型。

1. 设置环境变量 （二选一）​

  • 可以在执行脚本前设置环境变量：
export HF_ENDPOINT=https://hf-api.gitee.com
export HF_HOME=~/.cache/gitee-ai
  • 或者，在要执行的 python 代码 头部 粘贴以下代码:
python
import os
os.environ["HF_HOME"] = "~/.cache/gitee-ai"
os.environ["HF_ENDPOINT"] = "https://hf-api.gitee.com"

下载私有模型时，请使用 Gitee 私人令牌 设置环境变量 HF_TOKEN，然后再执行下载代码

export HF_TOKEN=<Gitee 私人令牌>

2. 下载模型或本地模型推理​

huggingface_hub:

python
from huggingface_hub import hf_hub_download
hf_hub_download("hf-models/sdxl-turbo", "sd_xl_turbo_1.0.safetensors", local_dir="./models")

transformers:

python
from transformers import pipeline
pipe = pipeline("text-generation", model="gpt2")

diffusers:

python
from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")

三. 下载统计​

在模型页面右侧，在线体验上方，我们提供了该模型最近 30 天的下载数据，下载包括本地克隆和仓库 ZIP 下载的总数。

alt text

模型的下载数是一个重要的受欢迎指标，所以我们在模型列表也提供了按下载数排序的功能，让用户可以快速找到最受欢迎的模型。

alt text