Update transformers/run_eval.py

sanchit-gandhi · web-flow · commit 2ea683ce97ed · 2024-02-07T17:19:43.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 (
             hasattr(asr_pipe.model.generation_config, "is_multilingual")

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 (`
`22`	`22`	`hasattr(asr_pipe.model.generation_config, "is_multilingual")`