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)   h/home/ubuntu/veenaModal/venv/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