Hugging Face's logo

Spaces:
derek-thomas
/
arabic-RAG
Build error

App Files Community
2
derek-thomas HF staff cbensimon HF staff commited on
Commit
ed7d1fc
·
1 Parent(s): 1a7a096

SentenceTransformer GPU device (#1)

Browse files

- SentenceTransformer GPU device (4edd64ff2ab451a38c3a32221e59fba79918a5a0)


Co-authored-by: Charles Bensimon <cbensimon@users.noreply.huggingface.co>

Files changed (1) hide show
  1. backend/semantic_search.py +1 -1
backend/semantic_search.py CHANGED
@@ -27,7 +27,7 @@ logger.info(f"Time taken to load LanceDB: {lancedb_loading_time:.6f} seconds")
27
  start_time = time.perf_counter()
28
 
29
  name = "sentence-transformers/paraphrase-multilingual-minilm-l12-v2"
30
- st_model_gpu = SentenceTransformer(name, device='mps')
31
  st_model_cpu = SentenceTransformer(name, device='cpu')
32
 
33
 
 
27
  start_time = time.perf_counter()
28
 
29
  name = "sentence-transformers/paraphrase-multilingual-minilm-l12-v2"
30
+ st_model_gpu = SentenceTransformer(name, device='cuda')
31
  st_model_cpu = SentenceTransformer(name, device='cpu')
32
 
33