huggingface · patrickvonplaten · Oct 10, 2023 · Oct 10, 2023 · Oct 10, 2023 · Oct 10, 2023
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -3077,6 +3077,7 @@
             "WHISPER_PRETRAINED_MODEL_ARCHIVE_LIST",
             "WhisperForAudioClassification",
             "WhisperForConditionalGeneration",
+            "WhisperForCausalLM"
             "WhisperModel",
             "WhisperPreTrainedModel",
         ]
@@ -6820,6 +6821,7 @@
             WHISPER_PRETRAINED_MODEL_ARCHIVE_LIST,
             WhisperForAudioClassification,
             WhisperForConditionalGeneration,
+            WhisperForCausalLM,
             WhisperModel,
             WhisperPreTrainedModel,
         )

diff --git a/src/transformers/generation/utils.py b/src/transformers/generation/utils.py