From 1773abd182d9392ee6547e09beb70586c1723242 Mon Sep 17 00:00:00 2001 From: Feiyang Date: Fri, 30 May 2025 00:21:37 +0000 Subject: [PATCH] revert to older model because gemini-embedding-001 doesn't support batch_predict at the moment. --- generative_ai/embeddings/batch_example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generative_ai/embeddings/batch_example.py b/generative_ai/embeddings/batch_example.py index c18ecf8d523..91be92de79b 100644 --- a/generative_ai/embeddings/batch_example.py +++ b/generative_ai/embeddings/batch_example.py @@ -39,7 +39,7 @@ def embed_text_batch() -> BatchPredictionJob: output_uri = OUTPUT_URI textembedding_model = language_models.TextEmbeddingModel.from_pretrained( - "gemini-embedding-001" + "textembedding-gecko@003" ) batch_prediction_job = textembedding_model.batch_predict(