From 5a1857ebcb9d3537fcf850321d2c30e512190e13 Mon Sep 17 00:00:00 2001 From: Addyk-24 Date: Mon, 13 Oct 2025 20:49:32 +0530 Subject: [PATCH] Fix: set forced_bos_token_id via generation_config --- examples/pytorch/translation/run_translation.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/pytorch/translation/run_translation.py b/examples/pytorch/translation/run_translation.py index e1d3c4ca387a..74e244994190 100755 --- a/examples/pytorch/translation/run_translation.py +++ b/examples/pytorch/translation/run_translation.py @@ -444,6 +444,9 @@ def main(): ) model.config.forced_bos_token_id = forced_bos_token_id + if hasattr(model, "generation_config") and model.generation_config is not None: + model.generation_config.forced_bos_token_id = forced_bos_token_id + # Get the language codes for input/target. source_lang = data_args.source_lang.split("_")[0] target_lang = data_args.target_lang.split("_")[0]