添加gpt-4-32k

2023-09-23 20:24:13 +08:00
parent 3672c97a06
commit 87f629bb37
2 changed files with 9 additions and 2 deletions
--- a/request_llm/bridge_all.py
+++ b/request_llm/bridge_all.py
@@ -125,7 +125,14 @@ model_info = {
        "tokenizer": tokenizer_gpt4,
        "token_cnt": get_token_num_gpt4,
    },
-
+    "gpt-4-32k": {
+        "fn_with_ui": chatgpt_ui,
+        "fn_without_ui": chatgpt_noui,
+        "endpoint": openai_endpoint,
+        "max_token": 32768,
+        "tokenizer": tokenizer_gpt4,
+        "token_cnt": get_token_num_gpt4,
+    },
    # azure openai
    "azure-gpt-3.5":{
        "fn_with_ui": chatgpt_ui,