跳到主要内容

下载模型

如果您想获取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)