Spaces:
Runtime error
Runtime error
derek-thomas
commited on
Commit
·
f6070bd
1
Parent(s):
5da9418
Updating GPU
Browse files
backend/semantic_search.py
CHANGED
|
@@ -24,7 +24,7 @@ start_time = time.perf_counter()
|
|
| 24 |
qd_retriever = EmbeddingRetriever(document_store=qd_document_store,
|
| 25 |
embedding_model="BAAI/bge-base-en-v1.5",
|
| 26 |
model_format="sentence_transformers",
|
| 27 |
-
use_gpu=
|
| 28 |
|
| 29 |
# Log the time taken to load the EmbeddingRetriever
|
| 30 |
retriever_loading_time = time.perf_counter() - start_time
|
|
|
|
| 24 |
qd_retriever = EmbeddingRetriever(document_store=qd_document_store,
|
| 25 |
embedding_model="BAAI/bge-base-en-v1.5",
|
| 26 |
model_format="sentence_transformers",
|
| 27 |
+
use_gpu=False)
|
| 28 |
|
| 29 |
# Log the time taken to load the EmbeddingRetriever
|
| 30 |
retriever_loading_time = time.perf_counter() - start_time
|