Spaces:
Sleeping
Sleeping
Tuchuanhuhuhu
commited on
Commit
•
cc35102
1
Parent(s):
85bb598
feat: 新增 GPT-3.5-16k 等新模型支持
Browse files- modules/presets.py +9 -1
modules/presets.py
CHANGED
@@ -60,11 +60,15 @@ APPEARANCE_SWITCHER = """
|
|
60 |
|
61 |
ONLINE_MODELS = [
|
62 |
"gpt-3.5-turbo",
|
|
|
63 |
"gpt-3.5-turbo-0301",
|
|
|
64 |
"gpt-4",
|
65 |
"gpt-4-0314",
|
|
|
66 |
"gpt-4-32k",
|
67 |
"gpt-4-32k-0314",
|
|
|
68 |
"川虎助理",
|
69 |
"川虎助理 Pro",
|
70 |
"xmchat",
|
@@ -105,11 +109,15 @@ for dir_name in os.listdir("models"):
|
|
105 |
|
106 |
MODEL_TOKEN_LIMIT = {
|
107 |
"gpt-3.5-turbo": 4096,
|
|
|
108 |
"gpt-3.5-turbo-0301": 4096,
|
|
|
109 |
"gpt-4": 8192,
|
110 |
"gpt-4-0314": 8192,
|
|
|
111 |
"gpt-4-32k": 32768,
|
112 |
-
"gpt-4-32k-0314": 32768
|
|
|
113 |
}
|
114 |
|
115 |
TOKEN_OFFSET = 1000 # 模型的token上限减去这个值,得到软上限。到达软上限之后,自动尝试减少token占用。
|
|
|
60 |
|
61 |
ONLINE_MODELS = [
|
62 |
"gpt-3.5-turbo",
|
63 |
+
"gpt-3.5-turbo-16k",
|
64 |
"gpt-3.5-turbo-0301",
|
65 |
+
"gpt-3.5-turbo-0613",
|
66 |
"gpt-4",
|
67 |
"gpt-4-0314",
|
68 |
+
"gpt-4-0613",
|
69 |
"gpt-4-32k",
|
70 |
"gpt-4-32k-0314",
|
71 |
+
"gpt-4-32k-0613",
|
72 |
"川虎助理",
|
73 |
"川虎助理 Pro",
|
74 |
"xmchat",
|
|
|
109 |
|
110 |
MODEL_TOKEN_LIMIT = {
|
111 |
"gpt-3.5-turbo": 4096,
|
112 |
+
"gpt-3.5-turbo-16k": 16384,
|
113 |
"gpt-3.5-turbo-0301": 4096,
|
114 |
+
"gpt-3.5-turbo-0613": 4096,
|
115 |
"gpt-4": 8192,
|
116 |
"gpt-4-0314": 8192,
|
117 |
+
"gpt-4-0613": 8192,
|
118 |
"gpt-4-32k": 32768,
|
119 |
+
"gpt-4-32k-0314": 32768,
|
120 |
+
"gpt-4-32k-0613": 32768
|
121 |
}
|
122 |
|
123 |
TOKEN_OFFSET = 1000 # 模型的token上限减去这个值,得到软上限。到达软上限之后,自动尝试减少token占用。
|