mini_whisper_model.py

# mini_whisper_model.py - Backwards compatibility alias
from configuration_mini_whisper import MiniWhisperConfig
from modeling_mini_whisper import (
    MiniWhisperForConditionalGeneration,
    MiniWhisperModel,
    MiniWhisperEncoder,
    MiniWhisperDecoder,
)

from transformers import CONFIG_MAPPING, MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING

CONFIG_MAPPING.register("mini_whisper", MiniWhisperConfig)
try:
    MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING.register(MiniWhisperConfig, MiniWhisperForConditionalGeneration)
except Exception:
    pass
MiniWhisperConfig.register_for_auto_class("AutoConfig")
MiniWhisperForConditionalGeneration.register_for_auto_class("AutoModelForSeq2SeqLM")

__all__ = [
    "MiniWhisperConfig",
    "MiniWhisperForConditionalGeneration",
    "MiniWhisperModel",
    "MiniWhisperEncoder",
    "MiniWhisperDecoder",
]