fix input
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -26,7 +26,7 @@ class PreTrainedPipeline():
|
|
26 |
- "feature_vector": A list of floats corresponding to the image embedding.
|
27 |
"""
|
28 |
|
29 |
-
batch_dict = self.tokenizer(inputs, max_length=512,
|
30 |
padding=True, truncation=True, return_tensors='pt')
|
31 |
with torch.no_grad():
|
32 |
outputs = self.model(**batch_dict)
|
|
|
26 |
- "feature_vector": A list of floats corresponding to the image embedding.
|
27 |
"""
|
28 |
|
29 |
+
batch_dict = self.tokenizer([inputs], max_length=512,
|
30 |
padding=True, truncation=True, return_tensors='pt')
|
31 |
with torch.no_grad():
|
32 |
outputs = self.model(**batch_dict)
|