| 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 | |