File size: 602 Bytes
ed084d7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
"""
BƯỚC 3: EMBEDDINGS – local & free
"""
from langchain_huggingface import HuggingFaceEmbeddings
EMBEDDING_MODEL = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
def get_embeddings():
print(f">>> Loading embedding model: {EMBEDDING_MODEL}")
embeddings = HuggingFaceEmbeddings(
model_name=EMBEDDING_MODEL,
model_kwargs={"device": "cpu"},
encode_kwargs={"normalize_embeddings": True},
)
print(">>> Embedding model loaded.\n")
return embeddings
if __name__ == "__main__":
emb = get_embeddings()
print(emb.embed_query("Test"))
|