MouadHsb commited on
Commit
c1911d8
·
1 Parent(s): ba0bad1

attempt to revert to embedding fix 2

Browse files
Files changed (1) hide show
  1. app/services/embedding_service.py +1 -2
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
- # Move model to CPU and eval mode
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