File size: 497 Bytes
57ea49c e694569 57ea49c e694569 57ea49c e694569 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from langchain_huggingface import HuggingFaceEmbeddings
from src.config.config import (
EMBEDDING_MODEL,
EMBEDDING_DEVICE,
NORMALIZE_EMBEDDINGS
)
def load_embeddings():
try:
return HuggingFaceEmbeddings(
model_name=EMBEDDING_MODEL,
model_kwargs={"device": EMBEDDING_DEVICE},
encode_kwargs={"normalize_embeddings": NORMALIZE_EMBEDDINGS}
)
except Exception as e:
raise RuntimeError(f"Failed to load embeddings: {e}")
|