diff --git a/generative_ai/multimodal_embedding_image.py b/generative_ai/multimodal_embedding_image.py index 53c1f6f9c417..d88ca2914746 100644 --- a/generative_ai/multimodal_embedding_image.py +++ b/generative_ai/multimodal_embedding_image.py @@ -28,7 +28,6 @@ def get_image_embeddings( location: str, image_path: str, contextual_text: Optional[str] = None, - dimension: int = 1408, ) -> MultiModalEmbeddingResponse: """Example of how to generate multimodal embeddings from image and text. @@ -37,8 +36,6 @@ def get_image_embeddings( location: Google Cloud Region, used to initialize vertexai image_path: Path to image (local or Google Cloud Storage) to generate embeddings for. contextual_text: Text to generate embeddings for. - dimension: Dimension for the returned embeddings. - https://cloud.google.com/vertex-ai/docs/generative-ai/embeddings/get-multimodal-embeddings#low-dimension """ vertexai.init(project=project_id, location=location) @@ -49,7 +46,6 @@ def get_image_embeddings( embeddings = model.get_embeddings( image=image, contextual_text=contextual_text, - dimension=dimension, ) print(f"Image Embedding: {embeddings.image_embedding}") print(f"Text Embedding: {embeddings.text_embedding}")