更改默认的LLM_MODEL

2023-07-21 15:35:17 +08:00 · 2023-07-21 15:35:17 +08:00 · 3b6819f091
parent 6ec1e56e03
commit 3b6819f091
1 changed files with 1 additions and 1 deletions
--- a/configs/model_config.py
+++ b/configs/model_config.py
@ -203,7 +203,7 @@ llm_model_dict = {
 }

 # LLM 名称
-LLM_MODEL = "fastchat-chatglm"
+LLM_MODEL = "chatglm-6b"
 # 量化加载8bit 模型
 LOAD_IN_8BIT = False
 # Load the model with bfloat16 precision. Requires NVIDIA Ampere GPU.