Spaces:
Sleeping
Sleeping
Adrian Cowham
commited on
Commit
•
ee8d7e4
1
Parent(s):
f0f775b
changed torch device to cpu
Browse files- 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': '
|
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,
|