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

codellama bug fix

Browse Source
This commit is contained in:
wangzaistone 2023-10-31 17:10:12 +08:00
parent 7619a16c16
commit 48539d7206
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/server/chat_adapter.py
View File

@ -218,7 +218,7 @@ class Llama2ChatAdapter(BaseChatAdpter):
class CodeLlamaChatAdapter(BaseChatAdpter): class CodeLlamaChatAdapter(BaseChatAdpter):
"""The model ChatAdapter for codellama .""" """The model ChatAdapter for codellama ."""
def match(self, model_path: str): def match(self, model_path: str):
return "codelama" in model_path.lower() return "codellama" in model_path.lower()
def get_conv_template(self, model_path: str) -> Conversation: def get_conv_template(self, model_path: str) -> Conversation:
return get_conv_template("codellama") return get_conv_template("codellama")