Spaces:

LunaticMaestro
/

rag

Sleeping

Deepak Sahu commited on Jan 22

Commit

32b22f3

1 Parent(s): 7f1eaf1

Update z_embedding.py

Files changed (1) hide show

z_embedding.py CHANGED Viewed

@@ -17,7 +17,7 @@ def get_embedding_model():
     embedding_model = HuggingFaceEmbeddings(
         model_name=EMBEDDING_MODEL_NAME,
-        multi_process=True,
         model_kwargs={"device": "cpu"},
         encode_kwargs={"normalize_embeddings": True},  # Set `True` for cosine similarity
     )

     embedding_model = HuggingFaceEmbeddings(
         model_name=EMBEDDING_MODEL_NAME,
+        multi_process=False,
         model_kwargs={"device": "cpu"},
         encode_kwargs={"normalize_embeddings": True},  # Set `True` for cosine similarity
     )