Adrian Cowham commited on
Commit
ee8d7e4
1 Parent(s): f0f775b

changed torch device to cpu

Browse files
Files changed (1) hide show
  1. src/core/embedding.py +1 -1
src/core/embedding.py CHANGED
@@ -51,7 +51,7 @@ def embed_files(
51
  files: List[File], embedding: str, vector_store: str, **kwargs
52
  ) -> FolderIndex:
53
  model_name = "BAAI/bge-small-en"
54
- model_kwargs = {'device': 'mps'}
55
  encode_kwargs = {'normalize_embeddings': True} # set True to compute cosine similarity
56
  model_norm = HuggingFaceBgeEmbeddings(
57
  model_name=model_name,
 
51
  files: List[File], embedding: str, vector_store: str, **kwargs
52
  ) -> FolderIndex:
53
  model_name = "BAAI/bge-small-en"
54
+ model_kwargs = {'device': 'cpu'}
55
  encode_kwargs = {'normalize_embeddings': True} # set True to compute cosine similarity
56
  model_norm = HuggingFaceBgeEmbeddings(
57
  model_name=model_name,