philschmid HF staff commited on
Commit
770e1d9
1 Parent(s): 5d9ef01

Update pipeline.py

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