From 4cc1065bd82219d7dd0cd6ada7ad3cf69900b6f0 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Fri, 2 Aug 2024 23:34:14 +0200 Subject: [PATCH] comment --- optimum/onnxruntime/modeling_seq2seq.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/optimum/onnxruntime/modeling_seq2seq.py b/optimum/onnxruntime/modeling_seq2seq.py index 03d51978c7f..d34916101a3 100644 --- a/optimum/onnxruntime/modeling_seq2seq.py +++ b/optimum/onnxruntime/modeling_seq2seq.py @@ -67,6 +67,8 @@ if check_if_transformers_greater("4.37.0"): + # starting from transformers v4.37.0, the whisper generation loop is implemented in the `WhisperGenerationMixin` + # and it implements many new features including short and long form generation, and starts with 2 init tokens from transformers.models.whisper.generation_whisper import WhisperGenerationMixin else: