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

feat(model): Support gemma model (#1187)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-02-23 12:37:13 +08:00
committed by GitHub
parent e0986198a6
commit f9d9f8faea
4 changed files with 57 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dbgpt/configs/model_config.py
View File

@@ -148,6 +148,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/google/gemma-7b-it
"gemma-7b-it": os.path.join(MODEL_PATH, "gemma-7b-it"),
# https://huggingface.co/google/gemma-2b-it
"gemma-2b-it": os.path.join(MODEL_PATH, "gemma-2b-it"),
}
EMBEDDING_MODEL_CONFIG = {