Update processing_aero.py
Browse files- processing_aero.py +3 -1
processing_aero.py
CHANGED
|
@@ -36,7 +36,9 @@ class AeroProcessorKwargs(ProcessingKwargs, total=False):
|
|
| 36 |
"text_kwargs": {
|
| 37 |
"padding": False,
|
| 38 |
},
|
| 39 |
-
"audio_kwargs": {
|
|
|
|
|
|
|
| 40 |
}
|
| 41 |
|
| 42 |
|
|
|
|
| 36 |
"text_kwargs": {
|
| 37 |
"padding": False,
|
| 38 |
},
|
| 39 |
+
"audio_kwargs": {
|
| 40 |
+
"return_tensors": "pt"
|
| 41 |
+
},
|
| 42 |
}
|
| 43 |
|
| 44 |
|