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 )é    )Ú	BaseModel)ÚListc                   @   s6   e Zd ZU eed< ee ed< ee ed< eed< dS )ÚRopeScalingConfigÚtypeÚlong_factorÚshort_factorÚ original_max_position_embeddingsN)Ú__name__Ú
__module__Ú__qualname__ÚstrÚ__annotations__r   ÚfloatÚint© r   r   úR/home/ubuntu/.local/lib/python3.10/site-packages/voxcpm/modules/minicpm4/config.pyr      s
   
 r   c                   @   sž   e Zd ZU eed< eed< eed< eed< eed< eed< eed< eed< eed	< eed
< eed< dZeed< eed< eed< eed< eed< dZ	eed< dS )ÚMiniCPM4ConfigÚbos_token_idÚeos_token_idÚhidden_sizeÚintermediate_sizeÚmax_position_embeddingsÚnum_attention_headsÚnum_hidden_layersÚnum_key_value_headsÚrms_norm_epsÚrope_scalingÚ
vocab_sizeTÚuse_mupÚ	scale_embÚdim_model_baseÚscale_depthÚ
rope_thetaNÚkv_channels)
r	   r
   r   r   r   r   r   r   Úboolr#   r   r   r   r   r      s$   
 r   N)Úpydanticr   Útypingr   r   r   r   r   r   r   Ú<module>   s    