feat: LVM - Released MultiModalEmbeddingModel to GA

PiperOrigin-RevId: 557054564
googleapis · Aug 15, 2023 · e99f366 · e99f366
1 parent fd5cb02
commit e99f366
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/tests/system/aiplatform/test_vision_models.py b/tests/system/aiplatform/test_vision_models.py
@@ -72,7 +72,7 @@ def test_image_q_and_a_model_ask_question(self):
     def test_multi_modal_embedding_model(self):
         aiplatform.init(project=e2e_base._PROJECT, location=e2e_base._LOCATION)
 
-        model = vision_models.MultiModalEmbeddingModel.from_pretrained(
+        model = ga_vision_models.MultiModalEmbeddingModel.from_pretrained(
             "multimodalembedding@001"
         )
         image = _create_blank_image()

diff --git a/tests/unit/aiplatform/test_vision_models.py b/tests/unit/aiplatform/test_vision_models.py
@@ -280,7 +280,7 @@ def test_image_embedding_model_with_only_text(self):
                 _IMAGE_EMBEDDING_PUBLISHER_MODEL_DICT
             ),
         ):
-            model = vision_models.MultiModalEmbeddingModel.from_pretrained(
+            model = ga_vision_models.MultiModalEmbeddingModel.from_pretrained(
                 "multimodalembedding@001"
             )
 

diff --git a/vertexai/vision_models/__init__.py b/vertexai/vision_models/__init__.py
@@ -18,10 +18,14 @@
     Image,
     ImageCaptioningModel,
     ImageQnAModel,
+    MultiModalEmbeddingModel,
+    MultiModalEmbeddingResponse,
 )
 
 __all__ = [
     "Image",
     "ImageCaptioningModel",
     "ImageQnAModel",
+    "MultiModalEmbeddingModel",
+    "MultiModalEmbeddingResponse",
 ]