use RetrievalQA instead of ChatVectorDBChain

This commit is contained in:
imClumsyPanda 2023-04-10 22:55:37 +08:00
parent 12ee17f3b3
commit 481aab97cb
1 changed files with 1 additions and 1 deletions

View File

@ -28,7 +28,7 @@ embedding_model_dict = {
llm_model_dict = { llm_model_dict = {
"chatglm-6b": "THUDM/chatglm-6b", "chatglm-6b": "THUDM/chatglm-6b",
"chatglm-6b-int4": "THUDM/chatglm-6b-int4", "chatglm-6b-int4": "THUDM/chatglm-6b-int4",
"chatglm-6b-int4-qe":"THUDM/chatglm-6b-int4-qe", "chatglm-6b-int4-qe": "THUDM/chatglm-6b-int4-qe",
} }
chatglm = ChatGLM() chatglm = ChatGLM()