Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
J
jinchat-server
概览
概览
详情
活动
周期分析
版本库
存储库
文件
提交
分支
标签
贡献者
分支图
比较
统计图
问题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
aigc-pioneer
jinchat-server
Commits
9bb7c219
Unverified
提交
9bb7c219
authored
4月 07, 2023
作者:
imClumsyPanda
提交者:
GitHub
4月 07, 2023
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #28 from lastrei/master
根据4.7日更新代码,解决冲突
上级
f239881b
1b110731
显示空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
6 行增加
和
1 行删除
+6
-1
knowledge_based_chatglm.py
knowledge_based_chatglm.py
+6
-1
没有找到文件。
knowledge_based_chatglm.py
浏览文件 @
9bb7c219
...
@@ -10,11 +10,16 @@ from langchain.vectorstores import FAISS
...
@@ -10,11 +10,16 @@ from langchain.vectorstores import FAISS
from
langchain.document_loaders
import
UnstructuredFileLoader
from
langchain.document_loaders
import
UnstructuredFileLoader
from
chatglm_llm
import
ChatGLM
from
chatglm_llm
import
ChatGLM
embedding_model_dict
=
{
"ernie-tiny"
:
"nghuyong/ernie-3.0-nano-zh"
,
"ernie-base"
:
"nghuyong/ernie-3.0-base-zh"
,
"text2vec"
:
"GanymedeNil/text2vec-large-chinese"
}
chatglm
=
ChatGLM
()
chatglm
=
ChatGLM
()
def
init_knowledge_vector_store
(
filepath
):
def
init_knowledge_vector_store
(
filepath
):
embeddings
=
HuggingFaceEmbeddings
(
model_name
=
"GanymedeNil/text2vec-large-chinese"
,
)
embeddings
=
HuggingFaceEmbeddings
(
model_name
=
embedding_model_dict
[
"text2vec"
]
,
)
loader
=
UnstructuredFileLoader
(
filepath
,
mode
=
"elements"
)
loader
=
UnstructuredFileLoader
(
filepath
,
mode
=
"elements"
)
docs
=
loader
.
load
()
docs
=
loader
.
load
()
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论