derek-thomas HF staff commited on
Commit
f6070bd
1 Parent(s): 5da9418

Updating GPU

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