Spaces:
Runtime error
Runtime error
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
|