gordonchan commited on
Commit
d0382a7
·
verified ·
1 Parent(s): e4d93e7

Update request_llms/bridge_all.py

Browse files
Files changed (1) hide show
  1. request_llms/bridge_all.py +16 -0
request_llms/bridge_all.py CHANGED
@@ -246,6 +246,22 @@ model_info = {
246
  "tokenizer": tokenizer_gpt35,
247
  "token_cnt": get_token_num_gpt35,
248
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
249
  }
250
 
251
  # -=-=-=-=-=-=- api2d 对齐支持 -=-=-=-=-=-=-
 
246
  "tokenizer": tokenizer_gpt35,
247
  "token_cnt": get_token_num_gpt35,
248
  },
249
+ "gemini-pro": {
250
+ "fn_with_ui": genai_ui,
251
+ "fn_without_ui": genai_noui,
252
+ "endpoint": None,
253
+ "max_token": 1024 * 32,
254
+ "tokenizer": tokenizer_gpt35,
255
+ "token_cnt": get_token_num_gpt35,
256
+ },
257
+ "gemini-pro-vision": {
258
+ "fn_with_ui": genai_ui,
259
+ "fn_without_ui": genai_noui,
260
+ "endpoint": None,
261
+ "max_token": 1024 * 32,
262
+ "tokenizer": tokenizer_gpt35,
263
+ "token_cnt": get_token_num_gpt35,
264
+ },
265
  }
266
 
267
  # -=-=-=-=-=-=- api2d 对齐支持 -=-=-=-=-=-=-