Skip to main content
HuggingFace 上的 BGE 模型最优秀的开源嵌入模型之一。 BGE 模型由北京智源人工智能研究院(BAAI)创建。BAAI 是一家从事 AI 研究与开发的私立非营利组织。
本笔记本展示如何通过 Hugging Face 使用 BGE 嵌入
pip install -qU  sentence_transformers
from langchain_community.embeddings import HuggingFaceBgeEmbeddings

model_name = "BAAI/bge-small-en"
model_kwargs = {"device": "cpu"}
encode_kwargs = {"normalize_embeddings": True}
hf = HuggingFaceBgeEmbeddings(
    model_name=model_name, model_kwargs=model_kwargs, encode_kwargs=encode_kwargs
)
注意,使用 model_name="BAAI/bge-m3" 时需要传入 query_instruction="",详见 FAQ BGE M3
embedding = hf.embed_query("hi this is harrison")
len(embedding)
384