diff --git a/src/transformers/models/whisper/generation_whisper.py b/src/transformers/models/whisper/generation_whisper.py index df9689b59788..436777d79bf7 100644 --- a/src/transformers/models/whisper/generation_whisper.py +++ b/src/transformers/models/whisper/generation_whisper.py @@ -1125,6 +1125,9 @@ def _set_return_outputs(return_dict_in_generate, return_token_timestamps, logpro return return_dict_in_generate def _set_return_timestamps(self, return_timestamps, is_shortform, generation_config): + if return_timestamps is None and hasattr(generation_config, "return_timestamps"): + return_timestamps = generation_config.return_timestamps + if not is_shortform: if return_timestamps is False: raise ValueError(