Update processing_aero.py
Browse files- processing_aero.py +3 -2
processing_aero.py
CHANGED
|
@@ -37,7 +37,6 @@ class AeroProcessorKwargs(ProcessingKwargs, total=False):
|
|
| 37 |
"padding": False,
|
| 38 |
},
|
| 39 |
"audio_kwargs": {
|
| 40 |
-
"return_tensors": "pt"
|
| 41 |
},
|
| 42 |
}
|
| 43 |
|
|
@@ -116,7 +115,9 @@ class AeroProcessor(ProcessorMixin):
|
|
| 116 |
tokenizer_init_kwargs=self.tokenizer.init_kwargs,
|
| 117 |
**kwargs,
|
| 118 |
)
|
| 119 |
-
|
|
|
|
|
|
|
| 120 |
|
| 121 |
if isinstance(text, str):
|
| 122 |
text = [text]
|
|
|
|
| 37 |
"padding": False,
|
| 38 |
},
|
| 39 |
"audio_kwargs": {
|
|
|
|
| 40 |
},
|
| 41 |
}
|
| 42 |
|
|
|
|
| 115 |
tokenizer_init_kwargs=self.tokenizer.init_kwargs,
|
| 116 |
**kwargs,
|
| 117 |
)
|
| 118 |
+
# Make sure no duplicate padding
|
| 119 |
+
if "padding" in output_kwargs["audio_kwargs"]:
|
| 120 |
+
output_kwargs["audio_kwargs"].pop("padding")
|
| 121 |
|
| 122 |
if isinstance(text, str):
|
| 123 |
text = [text]
|