ANYANTUDRE commited on
Commit
41f9e7f
·
1 Parent(s): f0dafb8

fixed small bugs

Browse files
Files changed (1) hide show
  1. goai_helpers/goai_stt2.py +7 -1
goai_helpers/goai_stt2.py CHANGED
@@ -51,7 +51,13 @@ def transcribe(
51
  if forced_decoder_ids:
52
  generate_kwargs["forced_decoder_ids"] = forced_decoder_ids
53
 
54
- output = pipe(inputs, batch_size=batch_size, padding=True, truncation=True, **generate_kwargs)
 
 
 
 
 
 
55
 
56
  transcription_text = output['text']
57
 
 
51
  if forced_decoder_ids:
52
  generate_kwargs["forced_decoder_ids"] = forced_decoder_ids
53
 
54
+ output = pipe(
55
+ inputs,
56
+ batch_size=batch_size,
57
+ #padding=True,
58
+ #truncation=True,
59
+ **generate_kwargs
60
+ )
61
 
62
  transcription_text = output['text']
63