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

feat(model): Support llama-3-sqlcoder-8b (#1559)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-05-25 17:26:04 +08:00
committed by GitHub
parent 83d7e9d82d
commit 82ed4611a5
2 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
dbgpt/configs/model_config.py
View File

@@ -197,6 +197,7 @@ LLM_MODEL_CONFIG = {
"phi-3-small-8k-instruct": os.path.join(MODEL_PATH, "Phi-3-small-8k-instruct"),
"phi-3-mini-128k-instruct": os.path.join(MODEL_PATH, "Phi-3-mini-128k-instruct"),
"phi-3-mini-4k-instruct": os.path.join(MODEL_PATH, "Phi-3-mini-4k-instruct"),
"llama-3-sqlcoder-8b": os.path.join(MODEL_PATH, "llama-3-sqlcoder-8b"),
}
EMBEDDING_MODEL_CONFIG = {