SentenceTransformer GPU device

#1
by cbensimon HF staff - opened
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