lite-whisper-large-v3-fast / configuration_lite_whisper.py
kamahori's picture
Upload LiteWhisperForConditionalGeneration
7d96429 verified
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