Bge 在 Hugging Face 上的集成

HuggingFace 上的 BGE 模型是最佳的开源嵌入模型之一。 BGE 模型由北京智源人工智能研究院 (BAAI)创建。BAAI 是一家从事人工智能研发的私营非营利组织。

本笔记本展示了如何通过 Hugging Face 使用 BGE Embeddings。

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=""，请参阅 BGE M3 常见问题解答。

embedding = hf.embed_query("hi this is harrison")
len(embedding)

将这些文档通过 MCP 连接到 Claude、VSCode 等，以获取实时答案。

在 GitHub 上编辑此页面或提交问题。