huggingface_hub优雅下载ChatGLM-6B模型

yf1ne

已于 2023-05-13 11:29:03 修改

阅读量5.7k

点赞数 7

CC 4.0 BY-SA版权

文章标签：自然语言处理人工智能服务器

于 2023-04-25 03:51:13 首次发布

本文指导如何使用huggingface_hub库从Hub下载ChatGLM-6B模型。首先，通过pip安装huggingface_hub，如果遇到网络问题，可选择清华源。然后创建目录并激活环境，最后使用Python和snapshot_download函数将模型保存至指定本地路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

安装huggingface_hub

pip install huggingface_hub

如果pip源无法连接, 可以尝试清华源

（临时方式）

pip install huggingface_hub -i https://pypi.tuna.tsinghua.edu.cn/simple

（永久方式）

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

在./ChatGLM-6B/下创建文件夹./ChatGLM-6B/chatglm-6b/用于存放本地模型

mkdir chatglm-6b

进入chatglm-6b环境, 进入python终端

conda activate chatglm-6b
python

调用huggingface_hub下载ChatGLM-6B模型到指定本地路径

from huggingface_hub import snapshot_download
snapshot_download(repo_id="THUDM/chatglm-6b", local_dir="./chatglm-6b/")

查看下载模型的文件目录

下载完成