diff --git a/configs/model_config.py b/configs/model_config.py
index c7e724d..9d66d37 100644
--- a/configs/model_config.py
+++ b/configs/model_config.py
@@ -23,7 +23,7 @@ llm_model_dict = {
 }
 
 # LLM model name
-LLM_MODEL = "chatyuan" #"chatglm-6b"
+LLM_MODEL = "chatglm-6b"
 
 # Use p-tuning-v2 PrefixEncoder
 USE_PTUNING_V2 = False