sima_utils.transformer.whisper_config
=====================================

.. py:module:: sima_utils.transformer.whisper_config


Classes
-------

.. autoapisummary::

   sima_utils.transformer.whisper_config.WhisperConfig


Module Contents
---------------

.. py:class:: WhisperConfig



   Base configuration with an update method.



   .. py:attribute:: model_type
      :type:  str
      :value: ''



   .. py:attribute:: d_model
      :type:  int
      :value: 768



   .. py:attribute:: encoder_attention_heads
      :type:  int
      :value: 12



   .. py:attribute:: encoder_layers
      :type:  int
      :value: 12



   .. py:attribute:: decoder_attention_heads
      :type:  int
      :value: 12



   .. py:attribute:: decoder_layers
      :type:  int
      :value: 12



   .. py:attribute:: max_source_positions
      :type:  int
      :value: 1500



   .. py:attribute:: max_target_positions
      :type:  int
      :value: 448



   .. py:attribute:: num_mel_bins
      :type:  int
      :value: 80



   .. py:attribute:: suppress_tokens
      :type:  list[int]
      :value: []



   .. py:attribute:: vocab_size
      :type:  int
      :value: 51865



   .. py:attribute:: activation_function
      :type:  str
      :value: 'gelu'



   .. py:method:: from_hf_config(model_path: pathlib.Path, model_cfg: dict) -> WhisperConfig
      :staticmethod:



   .. py:method:: load(model_cfg: dict) -> WhisperConfig
      :staticmethod:



   .. py:property:: encoder_head_dim
      :type: int



   .. py:property:: decoder_head_dim
      :type: int