github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-09 04:49:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(model): Support Yi-1.5 models (#1516)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-05-13 13:50:21 +08:00
committed by GitHub
parent c21cc8f583
commit 81fb56a67c
5 changed files with 50 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dbgpt/configs/model_config.py
View File

@@ -174,6 +174,10 @@ LLM_MODEL_CONFIG = {
# https://huggingface.co/01-ai/Yi-34B-Chat-4bits
"yi-34b-chat-4bits": os.path.join(MODEL_PATH, "Yi-34B-Chat-4bits"),
"yi-6b-chat": os.path.join(MODEL_PATH, "Yi-6B-Chat"),
# 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-34b-chat": os.path.join(MODEL_PATH, "Yi-1.5-34B-Chat"),
# 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