From b5f6f352a48aad5fe8027b37035fb7c8aab6ccb6 Mon Sep 17 00:00:00 2001 From: bog Date: Thu, 26 Dec 2024 18:26:16 +0100 Subject: [PATCH] fix: textembedding gecko 003 change from latest (#1572) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit # Description As described in the issue #1571 the gecko version can't load if it's specified as the latest, it's also not recommended by the documentation, I've changed it to the version 003. Fixes #1571 🦕 --- .../utils/intro_multimodal_rag_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gemini/use-cases/retrieval-augmented-generation/utils/intro_multimodal_rag_utils.py b/gemini/use-cases/retrieval-augmented-generation/utils/intro_multimodal_rag_utils.py index df5873dc29e..d9183c4ebfc 100644 --- a/gemini/use-cases/retrieval-augmented-generation/utils/intro_multimodal_rag_utils.py +++ b/gemini/use-cases/retrieval-augmented-generation/utils/intro_multimodal_rag_utils.py @@ -20,7 +20,7 @@ from vertexai.vision_models import Image as vision_model_Image from vertexai.vision_models import MultiModalEmbeddingModel -text_embedding_model = TextEmbeddingModel.from_pretrained("textembedding-gecko@latest") +text_embedding_model = TextEmbeddingModel.from_pretrained("textembedding-gecko@003") multimodal_embedding_model = MultiModalEmbeddingModel.from_pretrained( "multimodalembedding@001" )