feat: LVM - Released MultiModalEmbeddingModel to GA

PiperOrigin-RevId: 557054564

Author: Ark-kun

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()

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"
             )
 

vertexai/vision_models/__init__.py

@@ -18,10 +18,14 @@
     Image,
     ImageCaptioningModel,
     ImageQnAModel,
+    MultiModalEmbeddingModel,
+    MultiModalEmbeddingResponse,
 )
 
 __all__ = [
     "Image",
     "ImageCaptioningModel",
     "ImageQnAModel",
+    "MultiModalEmbeddingModel",
+    "MultiModalEmbeddingResponse",
 ]