From d525e543bddd57716196b9394d98677ab41c5d8d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9F=AF=E4=BB=95=E9=94=8B?= <12029064@zju.edu.cn> Date: Tue, 11 Mar 2025 10:01:08 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0gemini-2.0-flash?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- request_llms/bridge_all.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/request_llms/bridge_all.py b/request_llms/bridge_all.py index e911a765..43d61828 100644 --- a/request_llms/bridge_all.py +++ b/request_llms/bridge_all.py @@ -530,6 +530,15 @@ model_info = { "tokenizer": tokenizer_gpt35, "token_cnt": get_token_num_gpt35, }, + "gemini-2.0-flash": { + "fn_with_ui": genai_ui, + "fn_without_ui": genai_noui, + "endpoint": gemini_endpoint, + "has_multimodal_capacity": True, + "max_token": 1024 * 204800, + "tokenizer": tokenizer_gpt35, + "token_cnt": get_token_num_gpt35, + }, # cohere "cohere-command-r-plus": {