derek-thomas HF staff commited on
Commit
f9509ab
1 Parent(s): b888aa5

Fixing GPU for semantic_search.py

Browse files
Files changed (1) hide show
  1. backend/semantic_search.py +1 -1
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=False)
 
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)