comment

huggingface · Aug 2, 2024 · 4cc1065 · 4cc1065
1 parent 2f69a8a
commit 4cc1065
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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: