Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
J
jinchat-server
概览
概览
详情
活动
周期分析
版本库
存储库
文件
提交
分支
标签
贡献者
分支图
比较
统计图
问题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
aigc-pioneer
jinchat-server
Commits
aa266454
提交
aa266454
authored
5月 19, 2023
作者:
imClumsyPanda
浏览文件
操作
浏览文件
下载
差异文件
Merge remote-tracking branch 'origin/dev' into dev
上级
6c251bd2
65412e6c
隐藏空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
0 行增加
和
8 行删除
+0
-8
moss_llm.py
models/moss_llm.py
+0
-8
没有找到文件。
models/moss_llm.py
浏览文件 @
aa266454
from
abc
import
ABC
import
transformers
from
langchain.llms.base
import
LLM
from
typing
import
Optional
,
List
from
models.loader
import
LoaderCheckPoint
...
...
@@ -58,11 +57,6 @@ class MOSSLLM(BaseAnswer, LLM, ABC):
history
:
List
[
List
[
str
]]
=
[],
streaming
:
bool
=
False
,
generate_with_callback
:
AnswerResultStream
=
None
)
->
None
:
# Create the StoppingCriteriaList with the stopping strings
stopping_criteria_list
=
transformers
.
StoppingCriteriaList
()
# 定义模型stopping_criteria 队列,在每次响应时将 torch.LongTensor, torch.FloatTensor同步到AnswerResult
listenerQueue
=
AnswerResultQueueSentinelTokenListenerQueue
()
stopping_criteria_list
.
append
(
listenerQueue
)
if
len
(
history
)
>
0
:
history
=
history
[
-
self
.
history_len
:
-
1
]
if
self
.
history_len
>
0
else
[]
prompt_w_history
=
str
(
history
)
...
...
@@ -91,8 +85,6 @@ class MOSSLLM(BaseAnswer, LLM, ABC):
answer_result
=
AnswerResult
()
answer_result
.
history
=
history
answer_result
.
llm_output
=
{
"answer"
:
response
}
if
listenerQueue
.
listenerQueue
.
__len__
()
>
0
:
answer_result
.
listenerToken
=
listenerQueue
.
listenerQueue
.
pop
()
generate_with_callback
(
answer_result
)
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论