+
    œ°³jq#  ã                   ó  € ^ RI Ht ^RIHt ^RIHt ]! RR7      ] ! R R]4      4       4       t]! RR7      ] ! R R	]4      4       4       t]! R
R7      ] ! R R]4      4       4       t]! RR7      ] ! R R]4      4       4       t	. ROt
R# )é    )Ústrict)ÚPreTrainedConfig)Úauto_docstringznvidia/parakeet-ctc-1.1b)Ú
checkpointc                   ó    a a€ ] tR t^t oRtRtR.tRt^t^t	Rt
RtRtRt^	t^tRt^Pt^t^tR	tR
tR	tR	tR	tRtRtRtV 3R ltV3R ltRtVtV ;t # )ÚParakeetEncoderConfiga  
convolution_bias (`bool`, *optional*, defaults to `True`):
    Whether to use bias in convolutions of the conformer's convolution module.
conv_kernel_size (`int`, *optional*, defaults to 9):
    The kernel size of the convolution layers in the Conformer block.
subsampling_factor (`int`, *optional*, defaults to 8):
    The factor by which the input sequence is subsampled.
subsampling_conv_channels (`int`, *optional*, defaults to 256):
    The number of channels in the subsampling convolution layers.
num_mel_bins (`int`, *optional*, defaults to 80):
    Number of mel features.
subsampling_conv_kernel_size (`int`, *optional*, defaults to 3):
    The kernel size of the subsampling convolution layers.
subsampling_conv_stride (`int`, *optional*, defaults to 2):
    The stride of the subsampling convolution layers.
dropout_positions (`float`, *optional*, defaults to 0.0):
    The dropout ratio for the positions in the input sequence.
scale_input (`bool`, *optional*, defaults to `True`):
    Whether to scale the input embeddings.

Example:
```python
>>> from transformers import ParakeetEncoderModel, ParakeetEncoderConfig

>>> # Initializing a `ParakeetEncoder` configuration
>>> configuration = ParakeetEncoderConfig()

>>> # Initializing a model from the configuration
>>> model = ParakeetEncoderModel(configuration)

>>> # Accessing the model configuration
>>> configuration = model.config
```
Úparakeet_encoderÚpast_key_valuesé   i   ÚsiluTé   gš™™™™™¹?g        iˆ  g{®Gáz”?c                óJ   <€ V P                   V n        \        SV `  ! R/ VB  R # ©N© )Únum_attention_headsÚnum_key_value_headsÚsuperÚ__post_init__©ÚselfÚkwargsÚ	__class__s   &,€Út/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/models/parakeet/configuration_parakeet.pyr   Ú#ParakeetEncoderConfig.__post_init__T   s!   ø€ Ø#'×#;Ñ#;ˆÔ Ü‰ÒÑ' Ô'ó    c                óf  <€ V ^8„  d   Qh/ S[ ;R&   S[ ;R&   S[ ;R&   S[ ;R&   S[;R&   S[;R&   S[;R&   S[ ;R&   S[ ;R	&   S[ ;R
&   S[ ;R&   S[ ;R&   S[ ;R&   S[S[ ,          ;R&   S[S[ ,          ;R&   S[S[ ,          ;R&   S[S[ ,          ;R&   S[S[ ,          ;R&   S[ ;R&   S[;R&   S[;R&   # )é   Úhidden_sizeÚnum_hidden_layersr   Úintermediate_sizeÚ
hidden_actÚattention_biasÚconvolution_biasÚconv_kernel_sizeÚsubsampling_factorÚsubsampling_conv_channelsÚnum_mel_binsÚsubsampling_conv_kernel_sizeÚsubsampling_conv_strideÚdropoutÚdropout_positionsÚ	layerdropÚactivation_dropoutÚattention_dropoutÚmax_position_embeddingsÚscale_inputÚinitializer_range)ÚintÚstrÚboolÚfloat)ÚformatÚ__classdict__s   "€r   Ú__annotate__Ú"ParakeetEncoderConfig.__annotate__   sT  ø‡ ‚ ñR ÑñS ñT ÑñU ñV Ñ ñW ñX Ñ!ñY ñZ Ññ[ ñ\ Ññ] ñ^ Ñ!ñ_ ñ` Ñña ñb Ññc ñd  #Ñ(ñe ñf Ññg ñh #&Ñ)ñi ñj !Ñ$ñk ñl ‘S�[Ññm ñn ™s•{Ñ(ño ñp ‘s�{Ñ ñq ñr ¡�Ñ)ñs ñt ™s•{Ñ(ñu ñv !Ñ'ñw ñx Ññy ñz Ñ#ò{ r   )r   )!Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__Ú
model_typeÚkeys_to_ignore_at_inferencer   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__Ú__classcell__©r   r7   s   @@r   r   r      sŸ   ù‡ € ñ!ðF $€JØ#4Ð"5Ðà€KØÐØ ÐØ!ÐØ€JØ€NØ!ÐØÐØÐØ%(ÐØ€LØ()Ð Ø#$ÐØ€GØ%(ÐØ €IØ&)ÐØ%(ÐØ#'ÐØ€KØ#Ðõ(÷ … r   r   c                   ób   a a€ ] tR t^Yt oRtRtR]/tRtRt	Rt
RtRtV 3R	 ltV3R
 ltRtVtV ;t# )ÚParakeetCTCConfigaÈ  
ctc_loss_reduction (`str`, *optional*, defaults to `"mean"`):
    Specifies the reduction to apply to the output of `torch.nn.CTCLoss`. Only relevant when training an
    instance of [`ParakeetForCTC`].
ctc_zero_infinity (`bool`, *optional*, defaults to `True`):
    Whether to zero infinite losses and the associated gradients of `torch.nn.CTCLoss`. Infinite losses mainly
    occur when the inputs are too short to be aligned to the targets. Only relevant when training an instance
    of [`ParakeetForCTC`].
encoder_config (`Union[dict, ParakeetEncoderConfig]`, *optional*):
    The config object or dictionary of the encoder.

Example:

```python
>>> from transformers import ParakeetForCTC, ParakeetCTCConfig
>>> # Initializing a Parakeet configuration
>>> configuration = ParakeetCTCConfig()
>>> # Initializing a model from the configuration
>>> model = ParakeetForCTC(configuration)
>>> # Accessing the model configuration
>>> configuration = model.config
```
Úparakeet_ctcÚencoder_configi  ÚmeanTNr   c                ó  <€ \        V P                  \        4      '       d   \        R/ V P                  B V n        MV P                  f   \        4       V n        V P                  P                  V n        \
        SV `  ! R/ VB  R # r   ©Ú
isinstancerI   Údictr   r1   r   r   r   s   &,€r   r   ÚParakeetCTCConfig.__post_init__}   óg   ø€ Ü�d×)Ñ)¬4×0Ò0Ü"7Ñ"N¸$×:MÑ:MÑ"NˆDÕØ× Ñ Ò(Ü"7Ó"9ˆDÔØ!%×!4Ñ!4×!FÑ!FˆÔÜ‰ÒÑ' Ô'r   c                ó‚   <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   S[;R&   S[S[,          R,          ;R&   S[ R,          ;R&   # )r   Ú
vocab_sizeÚctc_loss_reductionÚctc_zero_infinityNrI   Úpad_token_id)r2   r3   r4   rN   r   )r6   r7   s   "€r   r8   ÚParakeetCTCConfig.__annotate__Y   s[   ø‡ ‚ ñ< Ññ= ñ> Ñ$ñ? ñ@ Ñ"ñA ñB Ñ+Õ+¨dÕ2Ñ9ñC ñD ˜•*Ñ#òE r   ©rI   r1   )r:   r;   r<   r=   r>   r?   r   Úsub_configsrR   rS   rT   rI   rU   r   rA   rB   rC   rD   rE   s   @@r   rG   rG   Y   sD   ù‡ € ñð0  €JØ#Ð%:Ð;€Kà€JØ$ÐØ"ÐØ59€NØ#€Lõ(÷I … r   rG   znvidia/parakeet-rnnt-0.6bc                   ór   a a€ ] tR t^†t oRtRtR]/tRtRt	^t
Rt^
tRt^tRtR	tV 3R
 ltV3R ltRtVtV ;t# )ÚParakeetRNNTConfiga¾  
decoder_hidden_size (`int`, *optional*, defaults to 640):
    Hidden size of the LSTM prediction network and joint network.
num_decoder_layers (`int`, *optional*, defaults to 2):
    Number of LSTM layers in the prediction network.
max_symbols_per_step (`int`, *optional*, defaults to 10):
    Maximum number of symbols to emit per encoder time step during greedy decoding.
encoder_config (`Union[dict, ParakeetEncoderConfig]`, *optional*):
    The config object or dictionary of the encoder.
blank_token_id (`int`, *optional*, defaults to 8192):
    Blank token id. Different from `pad_token_id` for RNN-T.

Example:
```python
>>> from transformers import ParakeetForRNNT, ParakeetRNNTConfig

>>> # Initializing a Parakeet RNN-T configuration
>>> configuration = ParakeetRNNTConfig()

>>> # Initializing a model from the configuration
>>> model = ParakeetForRNNT(configuration)

>>> # Accessing the model configuration
>>> configuration = model.config
```
Úparakeet_rnntrI   i   i€  ÚreluNi    Tc                ó  <€ \        V P                  \        4      '       d   \        R/ V P                  B V n        MV P                  f   \        4       V n        V P                  P                  V n        \
        SV `  ! R/ VB  R # r   rL   r   s   &,€r   r   Ú ParakeetRNNTConfig.__post_init__±   rP   r   c                ó¤   <€ V ^8„  d   Qh/ S[ ;R&   S[ ;R&   S[ ;R&   S[;R&   S[ ;R&   S[S[,          R,          ;R&   S[ ;R&   S[ ;R	&   S[;R
&   # )r   rR   Údecoder_hidden_sizeÚnum_decoder_layersr!   Úmax_symbols_per_stepNrI   rU   Úblank_token_idÚis_encoder_decoder)r2   r3   rN   r   r4   )r6   r7   s   "€r   r8   ÚParakeetRNNTConfig.__annotate__†   s—   ø‡ ‚ ñB ÑñC ñD Ñ"ñE ñF ÑñG ñH ÑñI ñJ Ñ"ñK ñL Ñ+Õ+¨dÕ2Ñ9ñM ñN ÑñO ñP ÑñQ ñR Ñ#òS r   rW   )r:   r;   r<   r=   r>   r?   r   rX   rR   r`   ra   r!   rb   rI   rU   rc   rd   r   rA   rB   rC   rD   rE   s   @@r   rZ   rZ   †   sZ   ù‡ € ñð6 !€JØ#Ð%:Ð;€Kà€JØ"ÐØÐØ€JØ "ÐØ59€NØ€LØ€NØ#Ðõ(÷W … r   rZ   znvidia/parakeet-tdt-0.6b-v3c                   ó8   a € ] tR t^ºt o RtRtRtV 3R ltRtV t	R# )ÚParakeetTDTConfigaâ  
A TDT (Token-and-Duration Transducer) extends the base RNN-T configuration [`ParakeetRNNTConfig`] with a
`durations` field: the joint network gains a duration head (its output width grows from `vocab_size` to
`vocab_size + len(durations)`), and during greedy decoding the encoder frame pointer advances by the
predicted duration rather than a fixed single frame.

decoder_hidden_size (`int`, *optional*, defaults to 640):
    Hidden size of the LSTM prediction network and joint network.
num_decoder_layers (`int`, *optional*, defaults to 2):
    Number of LSTM layers in the prediction network.
max_symbols_per_step (`int`, *optional*, defaults to 10):
    Maximum number of symbols to emit per encoder time step during greedy decoding.
durations (`list[int]`, *optional*, defaults to `[0, 1, 2, 3, 4]`):
    Token duration values that can be predicted. Each value represents how many frames a token or blank
    emission spans.
encoder_config (`Union[dict, ParakeetEncoderConfig]`, *optional*):
    The config object or dictionary of the encoder.
blank_token_id (`int`, *optional*, defaults to 8192):
    Blank token id. Different from `pad_token_id` for TDT.

Example:
```python
>>> from transformers import ParakeetForTDT, ParakeetTDTConfig

>>> # Initializing a Parakeet TDT configuration
>>> configuration = ParakeetTDTConfig()

>>> # Initializing a model from the configuration
>>> model = ParakeetForTDT(configuration)

>>> # Accessing the model configuration
>>> configuration = model.config
```
Úparakeet_tdtc                óZ   <€ V ^8„  d   Qh/ S[ S[,          S[S[R3,          ,          ;R&   # )r   .Ú	durations)Úlistr2   Útuple)r6   r7   s   "€r   r8   ÚParakeetTDTConfig.__annotate__º   s%   ø‡ ‚ ñN ‘C�y™5¡ c �?Õ*Ñ<òO r   r   N)r   é   r   é   é   )
r:   r;   r<   r=   r>   r?   rj   rA   rB   rC   )r7   s   @r   rg   rg   º   s   ø‡ € ñ!ðF  €JØ-<€I÷O ƒ r   rg   N)rG   r   rZ   rg   )Úhuggingface_hub.dataclassesr   Úconfiguration_utilsr   Úutilsr   r   rG   rZ   rg   Ú__all__r   r   r   Ú<module>ru      sÄ   ðõ /å 3Ý #ñ Ð5Ô6Øô?(Ð,ó ?(ó ó 7ð?(ñD Ð5Ô6Øô((Ð(ó ((ó ó 7ð((ñV Ð6Ô7Øô/(Ð)ó /(ó ó 8ð/(ñd Ð8Ô9Øô%=Ð*ó %=ó ó :ð%=òP d‚r   