Spaces:
Paused
Paused
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 |
|