feat: LVM - Released ImageQnAModel to GA

PiperOrigin-RevId: 557012696

Author: Ark-kun

tests/system/aiplatform/test_vision_models.py

@@ -59,7 +59,7 @@ def test_image_captioning_model_get_captions(self):
     def test_image_q_and_a_model_ask_question(self):
         aiplatform.init(project=e2e_base._PROJECT, location=e2e_base._LOCATION)
 
-        model = vision_models.ImageQnAModel.from_pretrained("imagetext")
+        model = ga_vision_models.ImageQnAModel.from_pretrained("imagetext")
         image = _create_blank_image()
         answers = model.ask_question(
             image=image,

tests/unit/aiplatform/test_vision_models.py

@@ -153,7 +153,7 @@ def test_get_captions(self):
                 _IMAGE_TEXT_PUBLISHER_MODEL_DICT
             ),
         ) as mock_get_publisher_model:
-            model = vision_models.ImageQnAModel.from_pretrained("imagetext@001")
+            model = ga_vision_models.ImageQnAModel.from_pretrained("imagetext@001")
 
         mock_get_publisher_model.assert_called_once_with(
             name="publishers/google/models/imagetext@001",

vertexai/vision_models/__init__.py

@@ -17,9 +17,11 @@
 from vertexai.vision_models._vision_models import (
     Image,
     ImageCaptioningModel,
+    ImageQnAModel,
 )
 
 __all__ = [
     "Image",
     "ImageCaptioningModel",
+    "ImageQnAModel",
 ]