命令行下载huggingface 受保护模型

发布时间:2024年01月11日

背景:

我要下载 huggingface 上的 meta 的 llama2 的模型到服务器上,这个模型是受保护的,需要先在 meta 的网站上提交申请,申请通过后,才能用和你申请时同邮箱绑定的hugging face账户进行下载这个模型。

方法:

确保你下载了 `huggingface-cli`

1、登陆

huggingface-cli login

此时会提示你没有 token,如下图所示:

然后点进去链接设置一个 token 并复制

然后运行下面代码,其中把$HUGGINGFACE_TOKEN换成你的token

huggingface-cli login --token $HUGGINGFACE_TOKEN

?登录成功后会出现下面提示:

2、下载?

使用?huggingface-cli download 命令,具体使用方法查看参考资料,或者huggingface-cli --help

由于我这里不想下载 llama2 的 safetensors,所以我的下载命令为:

huggingface-cli download meta-llama/Llama-2-7b-hf config.json generation_config.json pytorch_model-00001-of-00002.bin pytorch_model-00002-of-00002.bin pytorch_model.bin.index.json special_tokens_map.json tokenizer.json tokenizer.model tokenizer_config.json 

然后静待下载完毕即可,有进度条提示:

?

参考资料:

https://github.com/huggingface/huggingface_hub/blob/main/docs/source/en/guides/cli.md

文章来源:https://blog.csdn.net/qq_55621259/article/details/135527731
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。