From 481aab97cb48331d0826b1a916d8ed0ff68057d8 Mon Sep 17 00:00:00 2001
From: imClumsyPanda <littlepanda0716@gmail.com>
Date: Mon, 10 Apr 2023 22:55:37 +0800
Subject: [PATCH] use RetrievalQA instead of ChatVectorDBChain

---
 knowledge_based_chatglm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/knowledge_based_chatglm.py b/knowledge_based_chatglm.py
index 0d136d4..f00a8f2 100644
--- a/knowledge_based_chatglm.py
+++ b/knowledge_based_chatglm.py
@@ -28,7 +28,7 @@ embedding_model_dict = {
 llm_model_dict = {
     "chatglm-6b": "THUDM/chatglm-6b",
     "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()