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()