From b48e399e9e96f9c80529842fe505c0b061dd9466 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Fri, 21 Apr 2023 21:22:25 +0800 Subject: [PATCH] Update model_config.py --- configs/model_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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