Merge remote-tracking branch 'origin/short-rtf' into short-rtf

sanchit-gandhi · sanchit-gandhi · commit 1c73ee0d46ad · 2024-02-07T17:23:40.000Z
diff --git a/transformers/run_eval.py b/transformers/run_eval.py
@@ -16,7 +16,7 @@ def main(args):
     )
 
     if asr_pipe.model.can_generate():
-        gen_kwargs = {"max_new_tokens": 128}
+        gen_kwargs = {"max_new_tokens": 256}
         # for multilingual Whisper-checkpoints we see a definitive WER boost by setting the language and task args
         if getattr(asr_pipe.model.generation_config, "is_multilingual"):
             gen_kwargs["language"] = "en"

Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@ def main(args):`
`16`	`16`	`)`
`17`	`17`
`18`	`18`	`if asr_pipe.model.can_generate():`
`19`		`- gen_kwargs = {"max_new_tokens": 128}`
	`19`	`+ gen_kwargs = {"max_new_tokens": 256}`
`20`	`20`	`# for multilingual Whisper-checkpoints we see a definitive WER boost by setting the language and task args`
`21`	`21`	`if getattr(asr_pipe.model.generation_config, "is_multilingual"):`
`22`	`22`	`gen_kwargs["language"] = "en"`