add support for vision language model: google/paligemma

langchain-ai · May 14, 2024 · 4f69434 · 4f69434
1 parent e7bd7cd
commit 4f69434
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py b/libs/ai-endpoints/langchain_nvidia_ai_endpoints/_statics.py
@@ -171,6 +171,10 @@ class Model(BaseModel):
             "model_type": "chat",
             "model_name": "aisingapore/sea-lion-7b-instruct",
         },
+        "ai-google-paligemma": {
+            "model_type": "image_in",
+            "model_name": "google/paligemma",
+        },
     }
 )