lite-whisper-tiny-fast / configuration_lite_whisper.py
eyoel-gebre's picture
Upload LiteWhisperForConditionalGeneration
ac4b74d verified
raw
history blame contribute delete
307 Bytes
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