From 140d1ceafbf784a2dc387f7c26dd02af9df03a80 Mon Sep 17 00:00:00 2001 From: Fangyin Cheng Date: Mon, 20 May 2024 14:32:57 +0800 Subject: [PATCH] feat(model): Support Yi 1.5 16k models (#1543) --- dbgpt/configs/model_config.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/dbgpt/configs/model_config.py b/dbgpt/configs/model_config.py index d27d741f4..e530cd14b 100644 --- a/dbgpt/configs/model_config.py +++ b/dbgpt/configs/model_config.py @@ -178,7 +178,9 @@ LLM_MODEL_CONFIG = { # https://huggingface.co/01-ai/Yi-1.5-6B-Chat "yi-1.5-6b-chat": os.path.join(MODEL_PATH, "Yi-1.5-6B-Chat"), "yi-1.5-9b-chat": os.path.join(MODEL_PATH, "Yi-1.5-9B-Chat"), + "yi-1.5-9b-chat-16k": os.path.join(MODEL_PATH, "Yi-1.5-9B-Chat-16K"), "yi-1.5-34b-chat": os.path.join(MODEL_PATH, "Yi-1.5-34B-Chat"), + "yi-1.5-34b-chat-16k": os.path.join(MODEL_PATH, "Yi-1.5-34B-Chat-16K"), # https://huggingface.co/google/gemma-7b-it "gemma-7b-it": os.path.join(MODEL_PATH, "gemma-7b-it"), # https://huggingface.co/google/gemma-2b-it