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

refactor: Refactor proxy LLM (#1064)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-01-14 21:01:37 +08:00
committed by GitHub
parent a035433170
commit 22bfd01c4b
95 changed files with 2049 additions and 1294 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
dbgpt/model/adapter/model_adapter.py
View File

@@ -19,7 +19,7 @@ _IS_BENCHMARK = os.getenv("DB_GPT_MODEL_BENCHMARK", "False").lower() == "true"
_OLD_MODELS = [
"llama-cpp",
"proxyllm",
# "proxyllm",
"gptj-6b",
"codellama-13b-sql-sft",
"codellama-7b",
@@ -45,6 +45,7 @@ def get_llm_model_adapter(
# Import NewHFChatModelAdapter for it can be registered
from dbgpt.model.adapter.hf_adapter import NewHFChatModelAdapter
from dbgpt.model.adapter.proxy_adapter import ProxyLLMModelAdapter
new_model_adapter = get_model_adapter(
model_type, model_name, model_path, conv_factory