Skip to content

Commit 9e3e8f3

Browse files
authored
Update embedding.py
1 parent 1608578 commit 9e3e8f3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

generativeai/embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818

1919
def text_embedding():
2020
"""Text embedding with a Large Language Model."""
21-
model = TextEmbeddingModel.from_pretrained("google/embedding-gecko@001")
21+
model = TextEmbeddingModel.from_pretrained("textembedding-gecko@001")
2222
embeddings = model.get_embeddings(["What is life?"])
2323
for embedding in embeddings:
2424
vector = embedding.values

0 commit comments

Comments
 (0)