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

fix: lint

Browse Source
This commit is contained in:
csunny
2023-05-30 23:06:20 +08:00
parent b0e22eff05
commit 858cf472e2
4 changed files with 57 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/server/chat_adapter.py
View File

@@ -85,14 +85,14 @@ class CodeGenChatAdapter(BaseChatAdpter):
class GuanacoChatAdapter(BaseChatAdpter):
"""Model chat adapter for Guanaco """
"""Model chat adapter for Guanaco"""
def match(self, model_path: str):
return "guanaco" in model_path
def get_generate_stream_func(self):
from pilot.model.guanaco_llm import guanaco_generate_output
return guanaco_generate_output