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

feat(model): Support codegeex4-all-9b (#1720)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-07-14 20:39:38 +08:00
committed by GitHub
parent 125db534cb
commit e4abd56eeb
2 changed files with 22 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dbgpt/configs/model_config.py
View File

@@ -55,6 +55,8 @@ LLM_MODEL_CONFIG = {
# https://huggingface.co/THUDM/glm-4-9b-chat
"glm-4-9b-chat": os.path.join(MODEL_PATH, "glm-4-9b-chat"),
"glm-4-9b-chat-1m": os.path.join(MODEL_PATH, "glm-4-9b-chat-1m"),
# https://huggingface.co/THUDM/codegeex4-all-9b
"codegeex4-all-9b": os.path.join(MODEL_PATH, "codegeex4-all-9b"),
"guanaco-33b-merged": os.path.join(MODEL_PATH, "guanaco-33b-merged"),
"falcon-40b": os.path.join(MODEL_PATH, "falcon-40b"),
"gorilla-7b": os.path.join(MODEL_PATH, "gorilla-7b"),