Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
J
jinchat-server
概览
概览
详情
活动
周期分析
版本库
存储库
文件
提交
分支
标签
贡献者
分支图
比较
统计图
问题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
aigc-pioneer
jinchat-server
Commits
90fe40b2
提交
90fe40b2
authored
7月 19, 2023
作者:
hzg0601
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
debug for fastchat_openai_llm
上级
6f967a06
隐藏空白字符变更
内嵌
并排
正在显示
2 个修改的文件
包含
10 行增加
和
1 行删除
+10
-1
model_config.py
configs/model_config.py
+9
-0
fastchat_openai_llm.py
models/fastchat_openai_llm.py
+1
-1
没有找到文件。
configs/model_config.py
浏览文件 @
90fe40b2
...
...
@@ -155,6 +155,15 @@ llm_model_dict = {
"api_base_url"
:
"http://localhost:8000/v1"
,
# "name"修改为fastchat服务中的"api_base_url"
"api_key"
:
"EMPTY"
},
# 通过 fastchat 调用的模型请参考如下格式
"fastchat-chatglm-6b-int4"
:
{
"name"
:
"chatglm-6b-int4"
,
# "name"修改为fastchat服务中的"model_name"
"pretrained_model_name"
:
"chatglm-6b-int4"
,
"local_model_path"
:
None
,
"provides"
:
"FastChatOpenAILLMChain"
,
# 使用fastchat api时,需保证"provides"为"FastChatOpenAILLMChain"
"api_base_url"
:
"http://localhost:8001/v1"
,
# "name"修改为fastchat服务中的"api_base_url"
"api_key"
:
"EMPTY"
},
"fastchat-chatglm2-6b"
:
{
"name"
:
"chatglm2-6b"
,
# "name"修改为fastchat服务中的"model_name"
"pretrained_model_name"
:
"chatglm2-6b"
,
...
...
models/fastchat_openai_llm.py
浏览文件 @
90fe40b2
...
...
@@ -140,7 +140,7 @@ class FastChatOpenAILLMChain(RemoteRpcModel, Chain, ABC):
# create a chat completion
completion
=
openai
.
ChatCompletion
.
create
(
model
=
self
.
model_name
,
messages
=
build_message_list
(
prompt
)
messages
=
build_message_list
(
prompt
,
history
=
history
)
)
print
(
f
"response:{completion.choices[0].message.content}"
)
print
(
f
"+++++++++++++++++++++++++++++++++++"
)
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论