跳到主要内容

下载模型

如果您想获取CSGHub平台上的模型并下载至本地,当前我们支持通过Git、Web端下载、命令行下载以及SDK四种方式下载模型。

使用Git下载模型

  • 您可以通过HTTP下载模型:
git lfs install
git clone https://hub.opencsg.com/models/opencsg/test_model.git
  • 您也可以通过SSH下载模型:
git lfs install
git clone ssh://git@localhost:2222/models/opencsg/test_model.git

注:需要将SSH公钥添加到您的用户设置才能推送更改或访问私有存储库。 点击右上角“账号设置”,进入“SSH Keys”添加公钥。 SSH Key

使用Web端下载数据

在文件列表页点击下载按钮可直接下载文件。 Web端下载

使用命令行下载数据

使用命令行工具 csghub-cli 可以方便的下载数据,安装方法如下:

pip install csghub-sdk

使用方法如下:

export CSG_TOKEN=your_access_token

# download model
csghub-cli download demo/test_model

使用SDK下载数据

CSGHub SDK 提供了一个 Python Libaray,可以在代码中通过SDK下载数据。

示例代码如下:

from pycsghub.snapshot_download import snapshot_download
token = "your_access_token"

endpoint = "https://hub.opencsg.com"
repo_id = 'OpenCSG/csg-wukong-1B'
cache_dir = '/Users/hhwang/temp/'
result = snapshot_download(repo_id, cache_dir=cache_dir, endpoint=endpoint, token=token)

模型的多源同步

在CSGHub的开源版本中,可浏览远端传神社区的模型,进入某个项目,点击同步按钮,可快速将模型同步到本地服务器。 多源同步

点击同步按钮即可同步项目,按钮变为同步中(如下图)。 多源同步2

同步成功后,可以正常访问和下载项目(如下图)。 多源同步3

整个多源同步的介绍可参考如下视频