andrewluo commited on
Commit
dafed80
1 Parent(s): cc0dbd1

Update handler.py

Browse files
Files changed (1) hide show
  1. 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()