AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署

这篇具有很好参考价值的文章主要介绍了AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

前言

如果你要问我为什么直接部署ChatGLM2的模型?

因为当我在8月份在上海召开的全球人工智能大会上了解到清华-智谱发布的ChatGLM模型时,它已经发布了新的版本ChatGLM2,并且推理的效果提升了不少,那么本着只要最好的原则,我就直接上手先玩新版本了。

模型简介

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

部署的环境

作为AIGC方面的小白来说,我抱着非常天真的想法,想让它在我的工作笔记本上用i5的CPU去跑,至于为什么这么想,当然是因为我没有GPU,身边也没有其他的带显卡电脑

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

恰好,在腾讯云看到了GN7的显示优惠活动,60块钱15天,Nvidia T4的AI卡,直接斥巨资购买了15天

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

并且为了方便访问模型资源,我这里选择了新加坡的节点

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

软件环境

腾讯云的服务器,我选择了自己常用ubuntu镜像,同时此模型都是基于python语言,pytorch框架训练的模型,为了便于管理python版本库,使用anaconda来管理。

具体软件环境版本如下:

        conda 4.8.2

        python 3.9

        pytorch2.0.1

        cuda 11.7

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

具体的软件环境搭建过程,另起一篇去记录

代码工程及模型准备

1. github工程准备

git clone https://github.com/THUDM/ChatGLM2-6B.git

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

下载完后,进入到工程目录下 需要执行 pip install -r requirements.txt 安装需要的依赖

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

2. 模型权重准备

模型权重一般在huggingface上,但是现在huggingface被墙了,国内的同学需要科学上网才行,这里国内的小伙伴推荐去国内的modelscope下载模型权重

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

下载前需要安装git-lfs

sudo apt-get install git-lfs

使用git下载:

git lfs clone https://www.modelscope.cn/ZhipuAI/chatglm2-6b.git

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

运行

1.修改运行脚本cli_demo.sh

将模型路径修改为本地路径,即刚刚从modelscope下载的

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

2.运行推理

跑一个命令行的对话

python cli_demo.py

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch

同时,官方也提供了 网页版的web_demo,同样只需要修改模型路径为本地即可

体验

AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署,人工智能,AIGC,AIGC,人工智能,pytorch文章来源地址https://www.toymoban.com/news/detail-735557.html

到了这里,关于AIGC - ChatGLM大模型:ChatGLM2-6B模型推理部署的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包