Commit
•
770e1d9
1
Parent(s):
5d9ef01
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -27,4 +27,4 @@ class PreTrainedPipeline():
|
|
27 |
model_output = self.model(**encoded_input, return_dict=True)
|
28 |
embeddings = cls_pooling(model_output)
|
29 |
|
30 |
-
return {"vectors": embeddings[0].tolist()}
|
|
|
27 |
model_output = self.model(**encoded_input, return_dict=True)
|
28 |
embeddings = cls_pooling(model_output)
|
29 |
|
30 |
+
return {"vectors": [float(vec) for vec in embeddings[0].tolist()]}
|