
    0;ji	              	       \   d dl Z ddlmZ ddlmZmZmZmZmZm	Z	 ddl
mZmZmZmZmZ ddlmZmZ ddl
mZ ddlmZ e j        j        e j        j        e j        j        e j        j         ee j        d	e          e j        j        d
Zeeeeeee	e j        j        dZeeeeeeedZeedZ dS )    N   )Swish)LinearNoSubsamplingEmbedinigNoSubsamplingConv1dSubsampling2Conv2dSubsampling4Conv2dSubsampling6Conv2dSubsampling8)PositionalEncodingRelPositionalEncodingWhisperPositionalEncodingLearnablePositionalEncodingNoPositionalEncoding)MultiHeadedAttentionRelPositionMultiHeadedAttention)EspnetRelPositionalEncoding)LegacyLinearNoSubsamplingSiLU)hardtanhtanhreluseluswishgelu)linearlinear_legacyembedconv1d2conv2dconv2d6conv2d8paraformer_dummy)r   abs_posrel_posrel_pos_espnetno_posabs_pos_whisperembed_learnable_pe)selfattnrel_selfattn)!torchtransformer.activationr   transformer.subsamplingr   r   r   r   r	   r
   transformer.embeddingr   r   r   r   r   transformer.attentionr   r   r   r   nnHardtanhTanhReLUSELUgetattrGELUCOSYVOICE_ACTIVATION_CLASSESIdentityCOSYVOICE_SUBSAMPLE_CLASSESCOSYVOICE_EMB_CLASSESCOSYVOICE_ATTENTION_CLASSES     c/root/voice-cloning/.venv/lib/python3.11/site-packages/chatterbox/models/s3gen/utils/class_utils.py<module>r?      s    * * * * * *                            % % % % % % % % ? ? ? ? ? ? ? ? ? ? ? ? !HMHMHMWUXvu--HM     ".#! !!)	 	   !$1"05   %3    r=   