ANYANTUDRE
commited on
Commit
·
41f9e7f
1
Parent(s):
f0dafb8
fixed small bugs
Browse files
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(
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
|