radames
/

e5-large

Feature Extraction

Model card Files Files and versions Community

radames HF staff commited on Apr 12, 2023

Commit

1ac1ce8

•

1 Parent(s): a607b23

fix input

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -26,7 +26,7 @@ class PreTrainedPipeline():
                 - "feature_vector": A list of floats corresponding to the image embedding.
         """
-        batch_dict = self.tokenizer(inputs, max_length=512,
                                     padding=True, truncation=True, return_tensors='pt')
         with torch.no_grad():
             outputs = self.model(**batch_dict)

                 - "feature_vector": A list of floats corresponding to the image embedding.
         """
+        batch_dict = self.tokenizer([inputs], max_length=512,
                                     padding=True, truncation=True, return_tensors='pt')
         with torch.no_grad():
             outputs = self.model(**batch_dict)