diff --git a/tests/system/aiplatform/test_vision_models.py b/tests/system/aiplatform/test_vision_models.py
index b68f7f1688..b6ee690fbd 100644
--- 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
index 9592b4a3c5..00114602bf 100644
--- 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
index e62da8b01e..fb3a32fd5f 100644
--- 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",
 ]