Update processing_aero.py
Browse files- processing_aero.py +2 -2
processing_aero.py
CHANGED
|
@@ -13,7 +13,7 @@
|
|
| 13 |
# See the License for the specific language governing permissions and
|
| 14 |
# limitations under the License.
|
| 15 |
"""
|
| 16 |
-
Processor class for
|
| 17 |
"""
|
| 18 |
|
| 19 |
import math
|
|
@@ -132,7 +132,7 @@ class AeroProcessor(ProcessorMixin):
|
|
| 132 |
sampling_rate=sampling_rate,
|
| 133 |
return_attention_mask=True,
|
| 134 |
padding="max_length",
|
| 135 |
-
**output_kwargs,
|
| 136 |
)
|
| 137 |
audio_inputs["audio_attention_mask"] = audio_inputs.pop(
|
| 138 |
"attention_mask"
|
|
|
|
| 13 |
# See the License for the specific language governing permissions and
|
| 14 |
# limitations under the License.
|
| 15 |
"""
|
| 16 |
+
Processor class for Aero.
|
| 17 |
"""
|
| 18 |
|
| 19 |
import math
|
|
|
|
| 132 |
sampling_rate=sampling_rate,
|
| 133 |
return_attention_mask=True,
|
| 134 |
padding="max_length",
|
| 135 |
+
**output_kwargs["audio_kwargs"],
|
| 136 |
)
|
| 137 |
audio_inputs["audio_attention_mask"] = audio_inputs.pop(
|
| 138 |
"attention_mask"
|