huggingface_hub优雅下载ChatGLM-6B模型

作者：从前慢现在也慢 | 2024-03-02 06:45:34

踩

huggingface_hub

安装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/")

查看下载模型的文件目录

下载完成

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/179543