Fix the bug of oversized output tokens in gpt-3.5-turbo-1106.

yym68686 · Nov 9, 2023 · 09bd5fc · 09bd5fc
1 parent 0d62962
commit 09bd5fc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/chatgpt2api/chatgpt2api.py b/chatgpt2api/chatgpt2api.py
@@ -89,8 +89,10 @@ def __init__(
             if "gpt-4-32k" in engine
             else 7000
             if "gpt-4" in engine
+            else 4000
+            if "gpt-3.5-turbo-1106" in engine
             else 15000
-            if "gpt-3.5-turbo-16k" in engine or "gpt-3.5-turbo-1106" in engine
+            if "gpt-3.5-turbo-16k" in engine
             else 99000
             if "claude-2-web" in engine or "claude-2" in engine
             else 4000