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

Merge branch 'llm_fxp' into dev

Browse Source 
# Conflicts:
#	.env.template
#	pilot/out_parser/base.py
This commit is contained in:
yhjun1026
2023-06-01 15:35:01 +08:00
parent 6648a671df adfd9150d7
commit 96c516ab55
11 changed files with 158 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/server/chat_adapter.py
View File

@@ -91,8 +91,9 @@ class GuanacoChatAdapter(BaseChatAdpter):
return "guanaco" in model_path
def get_generate_stream_func(self):
# TODO
pass
from pilot.model.guanaco_llm import guanaco_generate_output
return guanaco_generate_output
class ProxyllmChatAdapter(BaseChatAdpter):
@@ -107,6 +108,7 @@ class ProxyllmChatAdapter(BaseChatAdpter):
register_llm_model_chat_adapter(VicunaChatAdapter)
register_llm_model_chat_adapter(ChatGLMChatAdapter)
register_llm_model_chat_adapter(GuanacoChatAdapter)
# Proxy model for test and develop, it's cheap for us now.
register_llm_model_chat_adapter(ProxyllmChatAdapter)