HuggingFace Hub模型上传与模型卡
请解释向HuggingFace Hub上传模型的完整流程。模型卡(Model Card)应该包含哪些信息?如何设置模型权限和审查?如何使用git-lfs管理大文件?如何发布模型的多个版本?
回答
我还是少年
上传模型到HF Hub:
from huggingface_hub import HfApi
api = HfApi()
# 创建仓库
api.create_repo(repo_id="username/my-model", repo_type="model")
# 上传文件
api.upload_folder(
folder_path="./my_model/",
repo_id="username/my-model",
)
或用git-lfs:
git lfs install
git clone https://huggingface.co/username/my-model
cd my-model
# 追踪大文件
huggingface-cli lfs-enable-largefiles .
cp ../my_model/* .
git add . && git commit -m "upload model"
git push
模型卡(Model Card):
---
language:
- zh
- en
license: apache-2.0
tags:
- text-generation
- transformers
pipeline_tag: text-generation
---
# 模型名称
## 模型描述
## 训练数据
## 评估结果
## 使用方式
## 局限性
权限管理:
- Public/Private设置
- gated:需要用户同意Term of Use
- 审查:触发安全审查(特定关键词/知名用户)
版本管理:
- Git commit作为版本
- 每调用一次save_pretrained()新commit
- 支持回滚到历史版本