Sentence Similarity
sentence-transformers
PyTorch
Transformers
English
t5
text-embedding
embeddings
information-retrieval
beir
text-classification
language-model
text-clustering
text-semantic-similarity
text-evaluation
prompt-retrieval
text-reranking
feature-extraction
English
Sentence Similarity
natural_questions
ms_marco
fever
hotpot_qa
mteb
Eval Results
Inference Endpoints
Update handler.py
Browse files- handler.py +0 -2
handler.py
CHANGED
@@ -23,7 +23,5 @@ class EndpointHandler():
|
|
23 |
instruction = data.pop("instruction",data)
|
24 |
text = data.pop("text", data)
|
25 |
inputs = [[s, instruction] for s in text]
|
26 |
-
if (self.device):
|
27 |
-
inputs = torch.tensor(inputs).to(self.device) # Move inputs to the GPU
|
28 |
embeddings = self.model.encode(inputs)
|
29 |
return embeddings.tolist()
|
|
|
23 |
instruction = data.pop("instruction",data)
|
24 |
text = data.pop("text", data)
|
25 |
inputs = [[s, instruction] for s in text]
|
|
|
|
|
26 |
embeddings = self.model.encode(inputs)
|
27 |
return embeddings.tolist()
|