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

feature: add guanaco adapter

Browse Source
This commit is contained in:
csunny 2023-05-29 17:59:50 +08:00
parent d114d134df
commit cfd72331ac
2 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/model/adapter.py
View File

@ -83,6 +83,10 @@ class ChatGLMAdapater(BaseLLMAdaper):
return model, tokenizer
class GuanacoAdapter(BaseLLMAdaper):
"""TODO Support guanaco"""
pass
class CodeGenAdapter(BaseLLMAdaper):
pass

10
pilot/server/chat_adapter.py
View File

@ -84,6 +84,16 @@ class CodeGenChatAdapter(BaseChatAdpter):
pass
class GuanacoAdapter(BaseChatAdpter):
"""Model chat adapter for Guanaco """
def match(self, model_path: str):
return "guanaco" in model_path
def get_generate_stream_func(self):
# TODO
pass
register_llm_model_chat_adapter(VicunaChatAdapter)
register_llm_model_chat_adapter(ChatGLMChatAdapter)