kcz358 commited on
Commit
e529cb3
·
verified ·
1 Parent(s): 91ff004

Update processing_aero.py

Browse files
Files changed (1) hide show
  1. 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
- output_kwargs["audio_kwargs"].pop("padding")
 
 
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]