
    .`i/                     V    d dl mZ d dlmZ  G d de          Z G d de          ZdS )    )PretrainedConfig)Qwen2_5OmniTextConfigc            +            e Zd ZdZ	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 d'dededeeeef         z  deeeef         z  dededededededed
z  dedededed ed!ed"ed#ed$ed%ef* fd&Z x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                 `   || _         || _        || _        || _        || _        || _        || _        || _        |	| _        |
| _	        || _
        || _        || _        || _        || _        || _        || _        || _        || _        || _        || _         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__s                          /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/transformers_utils/configs/midashenglm.pyr.   zDashengConfig.__init__"   s    2 #"$(,*
"" &",

$$&
""6"""""    )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__r1   s   @r2   r   r      sv       .J ,..0!$( # -.# .#.# .# %S/)	.#
 E#s(O+.# .# .# .# .# .# .# T\.# .# .# .#  !.#" #.#$ %.#& '.#( ).#* +.#, -.# .# .# .# .# .# .# .# .# .#r3   r   c            
       P     e Zd ZdZ	 	 	 	 d	dedz  dededz  dedz  f fdZ xZS )
MiDashengLMConfigmidashenglmN   audio_encoder_configsubsample_factortext_configaudio_token_idc                     t          di |pi | _        || _        |rt          di |nt                      | _        d | j        _        || _         t                      j        di | d S r+   )	r   rB   rC   r   rD   rope_parametersrE   r-   r.   )r/   rB   rC   rD   rE   r0   r1   s         r2   r.   zMiDashengLMConfig.__init__V   s     %2$Q$Q5I5OR$Q$Q! 0 )!00K000&(( 	
 ,0(,""6"""""r3   )NrA   NN)r4   r5   r6   r7   dictr8   r.   r<   r=   s   @r2   r?   r?   S   s        J -1 !#'%)# #"Tk# # D[	#
 d
# # # # # # # # # #r3   r?   N)transformersr   ;transformers.models.qwen2_5_omni.configuration_qwen2_5_omnir   r   r?   r,   r3   r2   <module>rK      s   2 * ) ) ) ) )     
1# 1# 1# 1# 1#$ 1# 1# 1#h# # # # #( # # # # #r3   