Skip to content

Commit 1333ea7

Browse files
committed
restore gemini 1.0 pro, which is really supposed to be supported.
1 parent 0406bac commit 1333ea7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

plugins/google/modelgauge/suts/google_genai_client.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ def safety_settings(self) -> Optional[Dict[HarmCategory, HarmBlockThreshold]]:
196196
return {harm: HarmBlockThreshold.BLOCK_LOW_AND_ABOVE for harm in GEMINI_HARM_CATEGORIES}
197197

198198

199-
gemini_models = ["gemini-1.5-flash", "gemini-1.5-pro"]
199+
gemini_models = ["gemini-1.0-pro", "gemini-1.5-flash", "gemini-1.5-pro"]
200200
for model in gemini_models:
201201
SUTS.register(GoogleGenAiDefaultSUT, model, model, InjectSecret(GoogleAiApiKey))
202202
SUTS.register(

0 commit comments

Comments
 (0)