如何优雅的从huggingface中下载模型?

  • Post author:
  • Post category:其他


我很懒,直接上代码

pip install huggingface_hub
from huggingface_hub import snapshot_download
snapshot_download(repo_id="THUDM/chatglm-6b", cache_dir="保存路径")

OK惹。(下载好的文件都不要删除,都有用,直接找到模型所在文件夹进行使用即可。



版权声明:本文为guojiajiajiu原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。