Hugging Face 上的 BGE
HuggingFace 上的 BGE 模型是最佳开源嵌入模型之一。BGE 模型由北京人工智能研究院 (BAAI)创建。
BAAI是一个从事人工智能研究和开发的非营利性私营组织。
此笔记本演示了如何通过Hugging Face使用BGE 嵌入
%pip install --upgrade --quiet 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
)
API 参考：HuggingFaceBgeEmbeddings
请注意，您需要为model_name="BAAI/bge-m3"传递query_instruction=""，请参阅BGE M3 常见问题。
embedding = hf.embed_query("hi this is harrison")
len(embedding)
384