Update src/core/embeddings.py

src/core/embeddings.py

@@ -1,17 +1,20 @@
-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}")
-
+import torch
+torch.set_default_device("cpu")
+
+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}")
+