From 3b6819f091227ea23a76d300b322dd44acfc49e3 Mon Sep 17 00:00:00 2001 From: hzg0601 Date: Fri, 21 Jul 2023 15:35:17 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=94=B9=E9=BB=98=E8=AE=A4=E7=9A=84LL?= =?UTF-8?q?M=5FMODEL?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 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 4a51c5c..ed277ee 100644 --- 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.