Spaces:
Build error
Build error
SentenceTransformer GPU device
#1
by
cbensimon
HF staff
- opened
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='
|
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 |
|