o
    
Û¾i/  ã                   @   s<   d dl mZ d dlmZ G dd„ deƒZG dd„ deƒZdS )é    )ÚPretrainedConfig)ÚQwen2_5OmniTextConfigc                +       s¾   e Zd ZdZ												
											d(dededeeeef B deeeef B dededededededed
B dedededed ed!ed"ed#ed$ed%ef*‡ fd&d'„Z‡  Z	S ))ÚDashengConfigÚmidashenglm_dasheng_encoderé   é  é   é   éô  é   ç      @TNç        ç     @¿@é   é    é€>  é@   Ú	embed_dimÚ	outputdimÚ
patch_sizeÚpatch_strideÚinput_channelsÚtarget_lengthÚdepthÚ	num_headsÚ	mlp_ratioÚqkv_biasÚinit_valuesÚ	drop_rateÚattn_drop_rateÚf_minÚf_maxÚcenterÚ
win_lengthÚ
hop_lengthÚsample_rateÚn_fftÚn_melsc                    s”   || _ || _|| _|| _|| _|| _|| _|| _|	| _|
| _	|| _
|| _|| _|| _|| _|| _|| _|| _|| _|| _|| _tƒ jdi |¤Ž d S ©N© )r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   ÚsuperÚ__init__)Úselfr   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   Úkwargs©Ú	__class__r)   ú_/home/ubuntu/.local/lib/python3.10/site-packages/vllm/transformers_utils/configs/midashenglm.pyr+   "   s,   zDashengConfig.__init__)r   r   r   r   r	   r
   r   r   r   TNr   r   r   r   Tr   r   r   r   r   )
Ú__name__Ú
__module__Ú__qualname__Ú
model_typeÚintÚtupleÚfloatÚboolr+   Ú__classcell__r)   r)   r.   r0   r      s„    êþýüûúùø	÷
öõôóòñðïîíìëêr   c                
       sH   e Zd ZdZ				d
dedB dededB dedB f‡ fdd	„Z‡  ZS )ÚMiDashengLMConfigÚmidashenglmNé   Úaudio_encoder_configÚsubsample_factorÚtext_configÚaudio_token_idc                    sX   t di |pi ¤Ž| _|| _|rtdi |¤Žntƒ | _d | j_|| _tƒ jdi |¤Ž d S r(   )	r   r=   r>   r   r?   Úrope_parametersr@   r*   r+   )r,   r=   r>   r?   r@   r-   r.   r)   r0   r+   V   s   ÿýzMiDashengLMConfig.__init__)Nr<   NN)r1   r2   r3   r4   Údictr5   r+   r9   r)   r)   r.   r0   r:   S   s    ûþýüûr:   N)Útransformersr   Ú;transformers.models.qwen2_5_omni.configuration_qwen2_5_omnir   r   r:   r)   r)   r)   r0   Ú<module>   s   4