From 48539d7206a754644306cd1541337ad37b2d1c4c Mon Sep 17 00:00:00 2001 From: wangzaistone Date: Tue, 31 Oct 2023 17:10:12 +0800 Subject: [PATCH] codellama bug fix --- pilot/server/chat_adapter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pilot/server/chat_adapter.py b/pilot/server/chat_adapter.py index 509305247..4b6dd0eed 100644 --- a/pilot/server/chat_adapter.py +++ b/pilot/server/chat_adapter.py @@ -218,7 +218,7 @@ class Llama2ChatAdapter(BaseChatAdpter): class CodeLlamaChatAdapter(BaseChatAdpter): """The model ChatAdapter for codellama .""" 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: return get_conv_template("codellama")