Spaces:
Running
Running
Clement Vachet
commited on
Commit
·
afc2218
1
Parent(s):
b347a6f
Comment out cpu embedding parameter
Browse files- indexing.py +1 -1
indexing.py
CHANGED
@@ -68,7 +68,7 @@ def create_db(splits, collection_name):
|
|
68 |
embedding = HuggingFaceEmbeddings(
|
69 |
model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
|
70 |
# model_name="sentence-transformers/all-MiniLM-L6-v2",
|
71 |
-
model_kwargs={"device": "cpu"},
|
72 |
# encode_kwargs={'normalize_embeddings': False}
|
73 |
)
|
74 |
chromadb.api.client.SharedSystemClient.clear_system_cache()
|
|
|
68 |
embedding = HuggingFaceEmbeddings(
|
69 |
model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
|
70 |
# model_name="sentence-transformers/all-MiniLM-L6-v2",
|
71 |
+
# model_kwargs={"device": "cpu"},
|
72 |
# encode_kwargs={'normalize_embeddings': False}
|
73 |
)
|
74 |
chromadb.api.client.SharedSystemClient.clear_system_cache()
|