当前位置:首页|资讯|ChatGLM|ChatGPT|GitHub

【AI-ChatGLM2】ChatGLM2-6B API部署

作者:sen1112发布时间:2023-09-30

ChatGLM2-6B API部署,为其他模型提示API接口

1.  为何需要API

当我们部署好ChatGLM后,就可以启动web_demo.py使用了。

但我们想结合LangChain时,就想到如果能把ChatGLM封装的像ChatGPT那样调用API,那样我们就可以把LangChain和ChatGLM解耦,就算以后我们要扩展或升级模型,只要重新封装新的API就行了。

2.  如何部署

开启的方法很简单,官方就给出了方法

GitHub - THUDM/ChatGLM2-6B: ChatGLM2-6B: An Open Bilingual Chat LLM | 开源双语对话语言模型

3.  实践过程

3.1.  启动API

我们按照官方给的方法,先执行以下命令安装依赖:

pip install fastapi uvicorn

再启动api.py

python api.py

如下图,启动过程跟web_demo.py区别不大


启动过程


3.2.  测试API

我们可以用官方给出的 用POST的方法,直接让编程助手(我这里使用讯飞)帮我们写出Python的POST请求,如下图运行没有什么问题

API使用

4.  总结

API部署的过程很简单,有了这一步,之后才能结合LangChain一起使用,我们编程的自由度更高。



Copyright © 2024 aigcdaily.cn  北京智识时代科技有限公司  版权所有  京ICP备2023006237号-1