Morgan Funtowicz
commited on
Commit
·
cb9be17
1
Parent(s):
38fa9fc
feat(embeddings): fix invalid nested array when allocating the embedding responses
Browse files- 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 = [
|
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:
|