From c926655a31a8d953df6edb33bd53ff8c49525756 Mon Sep 17 00:00:00 2001 From: Ruilin Huang Date: Fri, 7 Jun 2024 03:38:50 +0800 Subject: [PATCH] [whisper] don't overwrite return_timestamps when not passed to generate --- src/transformers/models/whisper/generation_whisper.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/transformers/models/whisper/generation_whisper.py b/src/transformers/models/whisper/generation_whisper.py index 4d60427d8bd0..bebfceeed1d2 100644 --- a/src/transformers/models/whisper/generation_whisper.py +++ b/src/transformers/models/whisper/generation_whisper.py @@ -1038,6 +1038,9 @@ def _set_return_outputs( @staticmethod def _set_return_timestamps(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(