Update model_config.py

2023-04-21 21:22:25 +08:00 · 2023-04-21 21:22:25 +08:00 · b48e399e9e
parent b9d8824fa5
commit b48e399e9e
1 changed files with 1 additions and 1 deletions
--- 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