sima_utils.transformer.model.whisper_decoder_init_model

Classes

WhisperDecoderInitModel

Implementation Whisper's decoder layers to process the input tokens and encoder outputs.

Module Contents

class sima_utils.transformer.model.whisper_decoder_init_model.WhisperDecoderInitModel

Implementation Whisper’s decoder layers to process the input tokens and encoder outputs.

layer_idx

Transformer layer index.

layer_idx: int
num_tokens: ClassVar[int] = 4
token_idx: ClassVar[int] = 0
gen_onnx_files()