From 3391155077af737615226a5b7d2e10841153b4b4 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Sat, 2 Sep 2023 15:33:57 +0800 Subject: [PATCH] update model_config.py.example --- configs/model_config.py.example | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/model_config.py.example b/configs/model_config.py.example index dddd4017..85b8b190 100644 --- a/configs/model_config.py.example +++ b/configs/model_config.py.example @@ -93,7 +93,7 @@ llm_model_dict = { "api_base_url": "http://127.0.0.1:8888/v1", "api_key": os.environ.get("ZHIPUAI_API_KEY"), "provider": "ChatGLMWorker", - "version": "chatglm_pro", + "version": "chatglm_pro", # 可选包括 "chatglm_lite", "chatglm_std", "chatglm_pro" }, } @@ -103,7 +103,7 @@ LLM_MODEL = "chatglm2-6b" # 历史对话轮数 HISTORY_LEN = 3 -# LLM 运行设备。可选项同Embedding 运行设备。 +# LLM 运行设备。设为"auto"会自动检测，也可手动设定为"cuda","mps","cpu"其中之一。 LLM_DEVICE = "auto" # 日志存储路径