AlexHung29629 commited on
Commit
12908ea
·
verified ·
1 Parent(s): 9acaf43

Update ultravox_processing.py

Browse files
Files changed (1) hide show
  1. ultravox_processing.py +1 -4
ultravox_processing.py CHANGED
@@ -157,10 +157,7 @@ class UltravoxProcessor(transformers.ProcessorMixin):
157
  data["audio_values"] = x.input_features
158
  else:
159
  data["audio_values"] = x.input_values
160
- if self.audio_padding == "max_length":
161
- data["audio_len"] = x.attention_mask.sum(-1) - 1
162
- else:
163
- data["audio_len"] = [data["audio_values"].shape[-1]]
164
 
165
  if text is not None:
166
  assert isinstance(
 
157
  data["audio_values"] = x.input_features
158
  else:
159
  data["audio_values"] = x.input_values
160
+ data["audio_len"] = x.attention_mask.sum(-1) - 1
 
 
 
161
 
162
  if text is not None:
163
  assert isinstance(