add Gemini Ultra for MM (run-llama#10463)

add Gemini Ultra
Arize-ai · Feb 5, 2024 · 0c0387e · 0c0387e
1 parent 703c346
commit 0c0387e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/llama_index/multi_modal_llms/gemini.py b/llama_index/multi_modal_llms/gemini.py
@@ -38,7 +38,10 @@
     pass
 
 # This lists the multi-modal models - see also llms.gemini for text models.
-GEMINI_MM_MODELS = ("models/gemini-pro-vision",)
+GEMINI_MM_MODELS = (
+    "models/gemini-pro-vision",
+    "models/gemini-ultra-vision",
+)
 
 
 class GeminiMultiModal(MultiModalLLM):