Spaces:
Runtime error
Runtime error
Commit
•
f9509ab
1
Parent(s):
b888aa5
Fixing GPU for semantic_search.py
Browse files
backend/semantic_search.py
CHANGED
@@ -7,4 +7,4 @@ qd_document_store = QdrantDocumentStore(path=str(proj_dir/'Qdrant'), index='RAGD
|
|
7 |
qd_retriever = EmbeddingRetriever(document_store=qd_document_store,
|
8 |
embedding_model="BAAI/bge-base-en-v1.5",
|
9 |
model_format="sentence_transformers",
|
10 |
-
use_gpu=
|
|
|
7 |
qd_retriever = EmbeddingRetriever(document_store=qd_document_store,
|
8 |
embedding_model="BAAI/bge-base-en-v1.5",
|
9 |
model_format="sentence_transformers",
|
10 |
+
use_gpu=True)
|