update bridge_all.py

awwaawwa · Jul 4, 2024 · 32e41aa · 32e41aa
1 parent c8cade3
commit 32e41aa
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/request_llms/bridge_all.py b/request_llms/bridge_all.py
@@ -177,6 +177,14 @@ def decode(self, *args, **kwargs):
         "tokenizer": tokenizer_gpt35,
         "token_cnt": get_token_num_gpt35,
     },
+    "gemma2:27b": {
+        "fn_with_ui": chatgpt_ui,
+        "fn_without_ui": chatgpt_noui,
+        "endpoint": openai_endpoint,
+        "max_token": 4096,
+        "tokenizer": tokenizer_gpt35,
+        "token_cnt": get_token_num_gpt35,
+    },
     "t1c/deepseek-math-7b-rl:latest": {
         "fn_with_ui": chatgpt_ui,
         "fn_without_ui": None,