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