sima_utils.transformer.whisper_config
Classes
Base configuration with an update method. |
Module Contents
- class sima_utils.transformer.whisper_config.WhisperConfig
Base configuration with an update method.
- model_type: str = ''
- d_model: int = 768
- encoder_attention_heads: int = 12
- encoder_layers: int = 12
- decoder_attention_heads: int = 12
- decoder_layers: int = 12
- max_source_positions: int = 1500
- max_target_positions: int = 448
- num_mel_bins: int = 80
- suppress_tokens: list[int] = []
- vocab_size: int = 51865
- activation_function: str = 'gelu'
- static from_hf_config(model_path: pathlib.Path, model_cfg: dict) WhisperConfig
- static load(model_cfg: dict) WhisperConfig
- property encoder_head_dim: int
- property decoder_head_dim: int