File size: 307 Bytes
7d96429 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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
|