Tuchuanhuhuhu commited on
Commit
cc35102
1 Parent(s): 85bb598

feat: 新增 GPT-3.5-16k 等新模型支持

Browse files
Files changed (1) hide show
  1. 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占用。