from transformers import WhisperConfig

class LiteWhisperConfig(WhisperConfig):
    model_type = "lite-whisper"

    def __init__(
        self, 
        low_rank_config: list[dict[str, int]] = None,
        **kwargs,
    ):
        super().__init__(**kwargs)
        self.low_rank_config = low_rank_config