跳到主要内容

如何在 Gitee 上托管 AI 模型

为了帮助开发者们更高效的学习和开发 AI 模型,Gitee 现已经支持了 AI 模型托管功能,为广大 AI 开发者提供高速稳定的托管服务,现在除了代码之外,你也可以在 Gitee 托管自己的 AI 模型了。

查看 AI 模型

AI 模型的管理界面和代码的管理界面风格统一,比较明显的区别在于在大文件(如 AI 模型文件)后会对其进行 LFS 文件的标注,并注明其所占空间大小,这样就能很方便地定位到 AI 模型文件。

LFS 文件大小

下载 AI 模型

通过网页下载

如果你想要下载仓库中的某个 AI 模型,就像下载代码仓库中的某个文件一样,点击其文件名并点击下载即可。

下载 LFS 文件

通过命令行下载

除网页外,你也可以通过命令行下载模型文件:

# 克隆仓库
git clone https://gitee.com/<your-gitee-repo-path>
cd <your-gitee-repo-path>

# 若已全局启动lfs,AI模型已全部自动下载完成
# 若未全局启动lfs,可手动下载指定模型
git lfs install

# 执行完成后,<model-file-path>的模型文件就可以直接使用了
git lfs pull -I <model-file-path>

通过 Git LFS 管理模型

你还可以使用 Git LFS 的方式对模型进行管理。在仓库标签页中的「管理」进入「Git LFS」选项,就可以查看该仓库内所有的 LFS 文件,包含其文件类型、大小等信息,并可以对它们进行下载和删除的操作。

管理 LFS

怎么把 AI 模型上传到 Gitee

那么,如果想在 Gitee 上托管 AI 模型该如何操作呢?流程和代码托管一样吗?

与代码托管类似,支持从 Hugging Face 导入和从本地推送两种方式进行 AI 模型的上传:

从 Hugging Face 导入

Hugging Face 是一家非常活跃的人工智能创业公司。它拥有一个活跃的人工智能社区。有超过 5000 多家机构都在 Hugging Face 的社区发布内容,包括 Google AI、Facebook AI、微软等,这也让它成为了目前开发者们最熟悉的人工智能技术社区。

现在你可以直接使用 Hugging Face 的模型地址来将其导入至 Gitee 中,其操作与代码仓库的导入是相同的:

  • 点击右上角头像旁的「+」,选择从外部仓库导入;
  • 在 Git 仓库 URL 中输入你想要导入的 Hugging Face 地址,Gitee 会为其自动设置名称及路径;
  • 点击「导入」,等待几分钟即可完成,同时后台开始同步 LFS 文件(速度取决于文件大小)。

导入 LFS

从本地推送

除了从 Hugging Face 导入外,由于 Gitee 全面支持 Git LFS 功能,所以 Gitee 同样支持从本地推送 AI 模型,其流程与代码推送略有不同的是,推送 AI 模型时需要加入 git lfs相关命令。

  • 推送 Hugging Face 的模型仓库
# 克隆仓库至本地
git clone https://huggingface.co/bert-base-uncased.git
cd bert-base-uncased

# 同步模型数据
git lfs fetch --all

# 推送到Gitee仓库
git lfs push https://gitee.com/<your-gitee-repo-path> --all
  • 推送自己的 AI 模型仓库
cd ai-models-project

# 初始化仓库
git init

# 添加remote
git remote add origin https://gitee.com/<your-gitee-repo-path>

// ...
// 创建ai模型,如:xxx.msgpack
// ...

# 追踪模型文件
git lfs track "*.msgpack"
# 提交追踪变更
git add .gitattributes
git commit -m "track *.msgpack"

# 提交模型
git add .
git commit -m "append ...."

# 推送模型
git lfs push origin --all
git push

想要获得更大的存储空间?

提示

目前该功能已经对所有付费企业开放,由于 AI 模型体积庞大,存储和流量成本较高,付费企业可以通过扩容 LFS 容量以存储更多的 AI 模型。

在企业控制台中,点击左下角的设置即可查看 Git LFS 配额使用情况,并可按需进行扩容操作。

LFS 扩容

如果需想要获取更多关于 LFS 扩容的信息,欢迎扫描下方二维码联系专属顾问,或前往 Gitee 企业版官网了解更多详情:https://gitee.com/enterprises。