社区供稿 | 快捷部署清华大模型 ChatGLM2-6B,一键搞定 HuggingFace Space 空间

07/03 20:30
阅读数 1.5K

分享一个简单的部署 ChatGLM2-6B 模型到 Hugging Face Space 的方法。





01


在这篇文章:终于部署成功!GPU 云环境搭建 ChatGLM2-6B 坎坷路中,用线上云算力的资源部署了一套 ChatGLM2-6B,最后显示一行提示:

“This share link expires in 72 hours. For free permanent hosting and GPU upgrades, run `gradio deploy` from Terminal to deploy to Spaces (https://huggingface.co/spaces)”

意思:“这个地址只有 72 小时,如获得永久的地址,需要运行 gradio deploy 命令来部署到 Hugging Face 的空间上。


开一个命令终端,试试部署到 Hugging Face



提示需要登录到 huggingface_hub 的 token。


打开 Hugging Face 网站,点击个人账户的设置部分(Profile)。



找到左边菜单中的“Access Tokens”,然后点击 “New token”。


截图中我的账号已经创建了好一个 token,见:没有硬件资源?免费使用Colab 搭建你自己的Stable Diffiusion在线模型!保姆级教程,这次直接复制使用。


粘贴 token 到终端窗口,对询问 “Add token as git credential? (Y/n)” 回复:“Y”,等了一会儿,给了我一闷棍:



别看满屏幕的字,有用的信息就两个:“Connection aborted.”,“Connection reset by peer”。


联想到之前部署的时候,帮助文档里 Hugging Face 赫然在列:



估计是无法直接连接了,怎么办?



02


想了半天,打开我的腾讯云账号,一咬牙,一跺脚,买了一个月的硅谷云服务器。



终端登录,git 克隆 ChatGLM2-2B 仓库,然后执行 gradio deploy 命令。 


出乎我的意外,居然也是同样的连接错误。看来虽然服务器地址不一样,还是执行了同样的网络策略。哭!


部署看起来简单,虽然只有一行命令,但是因为网络的连接问题,进展陷入了死胡同。


问问 ChatGLM,有没有什么好的解决方案。



看起来正确,感觉 ChatGLM2-6B 还真是不弱。国产之光,ChatGPT 国内产品的平替真不是白叫的。感兴趣的朋友,可以看 ChatGLM2-6B 初体验


最后想到,既然是执行发布命令,连接到 Hugging Face Space 去操作,其实把仓库克隆到本地电脑(有代理),再执行发布命令,是不是一样可行?



这个方案成功了,命令行下执行完 git 克隆命令,输入 token 之后,顺利进入下一步。部署命令 “gradio deploy” 会在 Hugging Face 上创建一个你的仓库空间:

Creating new Spaces Repo in '/Users/huangyong/Documents/ChatGLM2-6B'


请记住,在回答后续问题时,除了硬件配置以外,其他都可以回车采用默认值。

Enter Spaces hardware (cpu-basic, cpu-upgrade, t4-small, t4-medium, a10g-small, a10g-large, a100-large) [cpu-basic]:

这个地方输入:t4-medium,才能让 ChatGLM2-6B 跑起来

(看 ChatGLM-6B 官方文档有量化模型的设置,猜测可以让更少的硬件也能运行起来,我没仔细研究。)


第一次运行的时候,就按了回车,选择默认 cpu-basic 硬件配置,就没运行起来。


如果直接输入:t4-medium,会提示需要添加一个付款方式:You must add a payment method to your account 。



当时我没关注这条提示信息,直接用基本配置 cpu-basic 部署,结果就是失败。



RuntimeError:Found no NVIDIA driver。


如果硬件配置选择 t4-small,会卡在第5个 Checkpoint。



没有任何错误提示,也是卡这里郁闷了很久。也不报错,最后猜测是不是硬件资源里面内存不够,把空间升级到 medium,就能顺利运行起来了。



启动后,点右上角:“App”。



Hugging Face Space 提供一个永久的空间地址供访问:

https://hf.co/spaces/foghuang/ChatGLM2-6B


Space 上面还有很多好玩的,使用人工智能模型的应用,大家有兴趣可以点开试用。像下面的第一个就是文本和语音互转的应用。


https://hf.co/spaces


03


Hugging Face Space 部署总结:

  1. HF 注册账号;

  2. 添加 token;

  3. 添加支付方式,国内借记卡,信用卡均可;

  4. 能访问 HF 的环境;

  5. 默认情况下,硬件最小配置选择:t4-medium;

  6. 启动成功后,点 space 的 App,可以看到运行界面;

  7. HF 提供永久地址,可以让应用被外部用户访问。


本文转载自社区供稿项目,不代表官方立场。了解更多,请关注下面的公众号卡片:

如果你有好的文章希望通过我们的平台分享给更多人,请通过这个链接与我们联系: 

https://huggingface.link/tougao

本文分享自微信公众号 - Hugging Face(gh_504339124f0f)。
如有侵权,请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”,欢迎正在阅读的你也加入,一起分享。

展开阅读全文
加载中
点击引领话题📣 发布并加入讨论🔥
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部