Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
J
jinchat-server
概览
概览
详情
活动
周期分析
版本库
存储库
文件
提交
分支
标签
贡献者
分支图
比较
统计图
问题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
aigc-pioneer
jinchat-server
Commits
9f5d1afc
Unverified
提交
9f5d1afc
authored
5月 20, 2023
作者:
zhenkaivip
提交者:
GitHub
5月 20, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix bug : 1、历史最后一条丢失;2、上下两行连读造成歧义 (#415)
* fix _call bug * fix * fix bug : 上下两行连读造成歧义
上级
18d6b686
隐藏空白字符变更
内嵌
并排
正在显示
2 个修改的文件
包含
5 行增加
和
7 行删除
+5
-7
local_doc_qa.py
chains/local_doc_qa.py
+1
-1
chatglm_llm.py
models/chatglm_llm.py
+4
-6
没有找到文件。
chains/local_doc_qa.py
浏览文件 @
9f5d1afc
...
@@ -121,7 +121,7 @@ def similarity_search_with_score_by_vector(
...
@@ -121,7 +121,7 @@ def similarity_search_with_score_by_vector(
else
:
else
:
_id0
=
self
.
index_to_docstore_id
[
id
]
_id0
=
self
.
index_to_docstore_id
[
id
]
doc0
=
self
.
docstore
.
search
(
_id0
)
doc0
=
self
.
docstore
.
search
(
_id0
)
doc
.
page_content
+=
doc0
.
page_content
doc
.
page_content
+=
" "
+
doc0
.
page_content
if
not
isinstance
(
doc
,
Document
):
if
not
isinstance
(
doc
,
Document
):
raise
ValueError
(
f
"Could not find document for id {_id}, got {doc}"
)
raise
ValueError
(
f
"Could not find document for id {_id}, got {doc}"
)
doc_score
=
min
([
scores
[
0
][
id
]
for
id
in
[
indices
[
0
]
.
tolist
()
.
index
(
i
)
for
i
in
id_seq
if
i
in
indices
[
0
]]])
doc_score
=
min
([
scores
[
0
][
id
]
for
id
in
[
indices
[
0
]
.
tolist
()
.
index
(
i
)
for
i
in
id_seq
if
i
in
indices
[
0
]]])
...
...
models/chatglm_llm.py
浏览文件 @
9f5d1afc
...
@@ -69,19 +69,17 @@ class ChatGLM(LLM):
...
@@ -69,19 +69,17 @@ class ChatGLM(LLM):
history
:
List
[
List
[
str
]]
=
[],
history
:
List
[
List
[
str
]]
=
[],
streaming
:
bool
=
STREAMING
):
# -> Tuple[str, List[List[str]]]:
streaming
:
bool
=
STREAMING
):
# -> Tuple[str, List[List[str]]]:
if
streaming
:
if
streaming
:
for
inum
,
(
stream_resp
,
_
)
in
enumerate
(
self
.
model
.
stream_chat
(
history
+=
[[]]
for
stream_resp
,
_
in
self
.
model
.
stream_chat
(
self
.
tokenizer
,
self
.
tokenizer
,
prompt
,
prompt
,
history
=
history
[
-
self
.
history_len
:
-
1
]
if
self
.
history_len
>
0
else
[],
history
=
history
[
-
self
.
history_len
:
-
1
]
if
self
.
history_len
>
0
else
[],
max_length
=
self
.
max_token
,
max_length
=
self
.
max_token
,
temperature
=
self
.
temperature
,
temperature
=
self
.
temperature
,
top_p
=
self
.
top_p
,
top_p
=
self
.
top_p
,
)
)
:
):
torch_gc
()
torch_gc
()
if
inum
==
0
:
history
[
-
1
]
=
[
prompt
,
stream_resp
]
history
+=
[[
prompt
,
stream_resp
]]
else
:
history
[
-
1
]
=
[
prompt
,
stream_resp
]
yield
stream_resp
,
history
yield
stream_resp
,
history
torch_gc
()
torch_gc
()
else
:
else
:
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论