如果你在 Colab 上打开此 Notebook,你可能需要安装 LlamaIndex 🦙。
输入 [ ]
已复制!
%pip install llama-index-embeddings-fastembed
%pip install llama-index-embeddings-fastembed
输入 [ ]
已复制!
%pip install llama-index
%pip install llama-index
要使用此提供程序,需要安装 fastembed
包。
输入 [ ]
已复制!
%pip install fastembed
%pip install fastembed
支持的模型列表可以在这里找到。
输入 [ ]
已复制!
from llama_index.embeddings.fastembed import FastEmbedEmbedding
embed_model = FastEmbedEmbedding(model_name="BAAI/bge-small-en-v1.5")
from llama_index.embeddings.fastembed import FastEmbedEmbedding embed_model = FastEmbedEmbedding(model_name="BAAI/bge-small-en-v1.5")
100%|██████████| 76.7M/76.7M [00:18<00:00, 4.23MiB/s]
输入 [ ]
已复制!
embeddings = embed_model.get_text_embedding("Some text to embed.")
print(len(embeddings))
print(embeddings[:5])
embeddings = embed_model.get_text_embedding("Some text to embed.") print(len(embeddings)) print(embeddings[:5])
384 [-0.04166769981384277, 0.0018720313673838973, 0.02632238157093525, -0.036030545830726624, -0.014812108129262924]