目前国内外很多大模型都放到huggingface进行开源和维护,推荐研究大模型的读者学习一下。
本博客将会从huggingface的注册登录、获取许可权限和下载模型三个部分进行讲解,希望能帮助读者学会如何从huggingface中下载模型权重和配置文件。
一、注册登录
首先,在huggingface官网https://huggingface.co/
上注册一个自己的账号,
二、获取许可权限
当注册好账号后点击右上角自己的个人头像,显示界面如下图所示:
点击Access Tokens,进入如下界面后,点击create new token,创建一个新的许可token。
进入一个新的界面(如下图)后,就可以申请对应项目的token许可证了,这里以google的owlvit
零样本目标检测模型为例,首先取一个token权限的名字,这里以ours
这个名称为例,如下图所示:
接下来,将页面往下滑,滑到下图所示的位置,在扫描框中输入google/owlvit-base-patch32
,点击下拉框中对应的项目名称,如果还有其他想提取权限的项目,还可以继续添加多个项目,然后点击create token
。
点击后,会跳到一个悬浮的界面,如下图所示,这时候就说明token已经生成,这里可以点击copy
复制token,最好将其保存到一个文档中,因为这个token只有现在能看得完整,后面无论用什么方法都看不完整,也没办法复制,之后点击done
。
记得保存生成的token!!!
最后页面就会重新跳回到原来的界面,如下图所示,就会显示刚刚申请的这个token的信息,但是我们是没办法复制这个token值的。
点击后面的一个三点字符
,可以进行权限再编辑(但编辑之后就会重新生成新的token),取消token权限和删除token的操作。
三、下载模型和配置
这里以零样本目标检测模型google/owlvit-base-patch32
为例,在git bash界面上执行命令如下:
export HF_ENDPOINT=https://hf-mirror.com
huggingface-cli download --resume-download google/owlvit-base-patch32 --local-dir 保存模型路径 --local-dir-use-symlinks False --resume-download --token 申请的token
这样就可以将huggingface官网上的项目权重文件下载到设置的保存模型路径
了。
本博客编写至2024年11月20日。