ChatGLM2-6B通过命令行和网页demo访问

文章描述了作者在本地部署ChatGLM2-6BFP16模型后,通过命令行和网页界面进行对话应用的尝试,涉及到显存使用、错误调试和Streamlit工具的应用。作者还分享了Python学习路径和资源占用情况,最后提到将进行模型微调。
摘要由CSDN通过智能技术生成

在上一篇文章中已经在本地部署了ChatGLM2-6B FP16 / BF16版本(需要12.6G显存)。

这里再进行进一步的学习尝试和记录。

项目目录,pycharm最终形成的项目如下:

分别尝试以下几个demo:

1、命令行对话

cli_demo.py直接在命令行进行对话:

要修改就是本地模型的路径

这里“.”代表的是当前目录。注意这里一定要用“\\”否则会报错。我们运行一下试试:

能够正常进行对话!

我们查看一下CPU以及GPU的占用情况,大家可以参考:

内存占用4个多G(也有可能是我开启的时间有点久)。

关键是显存占用12.5G,利用率只有1%(因为这里只运行,并没有训练吧;之前试过用8G显存的1070m试过轻量版的,大概占用7G)。

2、网页界面对话web_demo.py

运行web_demo.py

也要修改模型加载路径:

右键运行

报错:

提示:Textbox 对象没有 style 属性

网上搜索得出:

我试试:

卸载:

安装:

再次运行web_demo.py

哇哦,确实可以用!开始对话:

介绍的python学习路径非常详细。

查看一下系统占用:

与命令行版本的差不多。

3、再运行web_demo2.py

同样修改路径:

运行:

一堆报错。但是我在开始的教程中看到:

嗯哼,试试。

安装成功

再次运行web_demo2.py

还是报错!

再去Git Hub上查看:

原来是采用命令:

streamlit run web_demo2.py

再pycharm终端中运行:

哇哦,没有问题!这里让输入email的,忘记截图了,网页自动打开:

开始对话:

完全没有问题,体验比刚刚的网页demo还要好一点。

查看一下资源占用情况:

跟之前的一模一样,同时电脑也运行流畅,欧克欧克!

接下来会进一步尝试学习微调大模型。

  • 7
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值