feat: update model names and configurations for Qwen2-Math-72B

awwaawwa · Sep 14, 2024 · 15b3e76 · 15b3e76
1 parent e3c7c4b
commit 15b3e76
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/core_functional.py b/core_functional.py
@@ -149,10 +149,10 @@ def get_core_functions():
             "Suffix":   "",
             "AutoClearHistory": True,
         },
-        "DeepSeekCoder解数学题":{
+        "Qwen2-Math-72B解数学题":{
             "Prefix":   r"",
-            "Suffix":   "\n 请通过逐步推理来解答问题，并把最终答案放置于\\boxed{} 中。",
-            "ModelOverride": "deepseek-coder",
+            "Suffix":   "\n  Please solve the problem step by step and put the final answer in \\boxed{}.",
+            "ModelOverride": "Qwen/Qwen2-Math-72B-Instruct",
             "AutoClearHistory": True,
         },
         "生成代码Mathematica代码，解决数学问题": {

diff --git a/request_llms/bridge_all.py b/request_llms/bridge_all.py
@@ -112,6 +112,15 @@ def decode(self, *args, **kwargs):
 # -=-=-=-=-=-=- 以下这部分是最早加入的最稳定的模型 -=-=-=-=-=-=-
 model_info = {
     # OhMyGPT 兼容接口
+    "Qwen/Qwen2-Math-72B-Instruct": {  # 16k
+        "can_multi_thread": True,
+        "fn_with_ui": chatgpt_ui,
+        "fn_without_ui": chatgpt_noui,
+        "endpoint": openai_endpoint,
+        "max_token": 32000,
+        "tokenizer": tokenizer_gpt35,
+        "token_cnt": get_token_num_gpt35,
+    },
     "deepseek-chat": {  # 16k
         "can_multi_thread": True,
         "fn_with_ui": chatgpt_ui,