idZddlmZddlmZddlmZedeGdd eZedeGd d eZd d gZ y ) zParakeet model configuration.)strict)PreTrainedConfig)auto_docstringznvidia/parakeet-ctc-1.1b) checkpointcreZdZUdZdZdgZdZeed<dZ eed<dZ eed <d Z eed <d Z e ed <dZeed<dZeed<dZeed<dZeed<dZeed<dZeed<dZeed<dZeed<dZeezed<dZeezed<dZeezed <dZeezed!<dZeezed"<d#Zeed$<dZeed%<d&Zeed'<fd(Z xZ!S))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 ``` This configuration class is based on the ParakeetEncoder architecture from NVIDIA NeMo. You can find more details and pre-trained models at [nvidia/parakeet-ctc-1.1b](https://huggingface.co/nvidia/parakeet-ctc-1.1b). parakeet_encoderpast_key_values hidden_sizenum_hidden_layersnum_attention_headsiintermediate_sizesilu hidden_actTattention_biasconvolution_bias conv_kernel_sizesubsampling_factorsubsampling_conv_channelsP num_mel_binsrsubsampling_conv_kernel_sizesubsampling_conv_strideg?dropoutgdropout_positions layerdropactivation_dropoutattention_dropoutimax_position_embeddings scale_inputg{Gz?initializer_rangec F|j|_t| di|yN)rnum_key_value_headssuper __post_init__selfkwargs __class__s /var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/models/parakeet/configuration_parakeet.pyr.z#ParakeetEncoderConfig.__post_init__Xs!#'#;#;  '')"__name__ __module__ __qualname____doc__ model_typekeys_to_ignore_at_inferencer int__annotations__rrrrstrrboolrrrrrrr r!floatr"r#r$r%r&r'r(r. __classcell__r2s@r3r r s$L$J#4"5Ks  !s!JND!d!c%(s(L#() #)#$S$GUS[%(us{( Ius{ &) )%(us{(#'S'K#u#((r4r ceZdZUdZdZdeiZdZee d<dZ e e d<dZ e e d <d Zeezd ze d<d Zed ze d <fd ZxZS)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_ctcencoder_configi vocab_sizemeanctc_loss_reductionTctc_zero_infinityNr pad_token_idc t|jtrtdi|j|_n|jt|_|jj|_t |di|yr*) isinstancerEdictr r(r-r.r/s r3r.zParakeetCTCConfig.__post_init__sd d))4 0"7"N$:M:M"ND   ("7"9D !%!4!4!F!F ''r4)r5r6r7r8r9r sub_configsrFr;r<rHr=rIr>rErMrrJr.r@rAs@r3rCrC]sg0 J#%:;KJ$$"t"59ND++d29#L#*#((r4rCN) r8huggingface_hub.dataclassesrconfiguration_utilsrutilsrr rC__all__r+r4r3rSsy$.3#56B(,B(7B(J56(((((7((V  7 8r4