Morgan Funtowicz commited on
Commit
cb9be17
·
1 Parent(s): 38fa9fc

feat(embeddings): fix invalid nested array when allocating the embedding responses

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -56,7 +56,7 @@ class SentenceTransformerHandler(Handler):
56
  tokens = self._model.tokenize(request.input)
57
  vectors = self._model.encode(request.input)
58
 
59
- embeddings = [[None] * len(request)]
60
  if not request.is_batched:
61
  embeddings[0] = Embedding(index=0, embedding=vectors.tolist())
62
  else:
 
56
  tokens = self._model.tokenize(request.input)
57
  vectors = self._model.encode(request.input)
58
 
59
+ embeddings = [None] * len(request)
60
  if not request.is_batched:
61
  embeddings[0] = Embedding(index=0, embedding=vectors.tolist())
62
  else: