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)
|