Spaces:
Sleeping
Sleeping
MouadHsb
commited on
Commit
·
c1911d8
1
Parent(s):
ba0bad1
attempt to revert to embedding fix 2
Browse files
app/services/embedding_service.py
CHANGED
@@ -25,8 +25,7 @@ class EmbeddingService:
|
|
25 |
self.tokenizer = AutoTokenizer.from_pretrained(model_name)
|
26 |
self.model = AutoModel.from_pretrained(model_name)
|
27 |
|
28 |
-
|
29 |
-
self.model = self.model.to("cpu")
|
30 |
self.model.eval()
|
31 |
|
32 |
# Get embedding dimension from model config
|
|
|
25 |
self.tokenizer = AutoTokenizer.from_pretrained(model_name)
|
26 |
self.model = AutoModel.from_pretrained(model_name)
|
27 |
|
28 |
+
|
|
|
29 |
self.model.eval()
|
30 |
|
31 |
# Get embedding dimension from model config
|