update

ShishirPatil · Oct 28, 2024 · 0f10942 · 0f10942
1 parent 132eeeb
commit 0f10942
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/berkeley-function-call-leaderboard/bfcl/model_handler/oss_model/minicpm.py b/berkeley-function-call-leaderboard/bfcl/model_handler/oss_model/minicpm.py
@@ -1,5 +1,6 @@
 from bfcl.model_handler.oss_model.base_oss_handler import OSSHandler
 
+
 class MiniCPMHandler(OSSHandler):
     def __init__(self, model_name, temperature) -> None:
         super().__init__(model_name, temperature)
@@ -11,7 +12,9 @@ def _format_prompt(self, messages, function):
         formatted_prompt = ""
 
         for message in messages:
-            formatted_prompt += f"<|im_start|>{message['role']}\n{message['content']}<|im_end|>\n"
+            formatted_prompt += (
+                f"<|im_start|>{message['role']}\n{message['content']}<|im_end|>\n"
+            )
 
         formatted_prompt += f"<|im_start|>assistant\n"
 

diff --git a/berkeley-function-call-leaderboard/bfcl/model_handler/oss_model/minicpm_fc.py b/berkeley-function-call-leaderboard/bfcl/model_handler/oss_model/minicpm_fc.py
@@ -29,7 +29,6 @@
 class MiniCPMFCHandler(OSSHandler):
     def __init__(self, model_name, temperature) -> None:
         super().__init__(model_name, temperature)
-        self.model_name_huggingface = "/user/panyinxu/models/minicpm3-4b"
         self.stop_token_ids = [2, 73440]
 
     @overrides