§
    .`ƒiŠ  ã                   ó6   — d dl Z d dlmZ  G d„ de¦  «        ZdS )é    N)ÚPretrainedConfigc                   óØ   ‡ — e Zd ZdZ	 	 	 	 	 	 	 dd	ed
ededededededz  fˆ fd„Zedeej	        z  dd fd„¦   «         Z
ed„ ¦   «         Zed„ ¦   «         Zej        defd„¦   «         Zˆ xZS )ÚMedusaConfigÚmedusaé   é}  é   é   é@   é
   NÚhidden_sizeÚ
vocab_sizeÚ	num_headsÚnum_hidden_layersÚ	max_pathsÚtopkÚtruncated_vocab_sizec                 óà   •— || _         || _        || _        || _        || _        || _        t          d¦  «        | _        |€|n|| _        d|vrdg|d<    t          ¦   «         j
        di |¤Ž d S )Ni   ÚarchitecturesÚMedusaModel© )r   r   r   r   r   r   ÚintÚmax_seq_lenr   ÚsuperÚ__init__)
Úselfr   r   r   r   r   r   r   ÚkwargsÚ	__class__s
            €úz/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/transformers_utils/configs/medusa.pyr   zMedusaConfig.__init__   s   ø€ ð 'ˆÔØ$ˆŒØ"ˆŒØ!2ˆÔØ"ˆŒØˆŒ	Ý˜u™:œ:ˆÔà.Ð6ˆJˆJÐ<Pð 	Ô!ð  &Ð(Ð(Ø'4 oˆF?Ñ#à‰ŒÔÐ"Ð"˜6Ð"Ð"Ð"Ð"Ð"ó    Úpretrained_model_name_or_pathÚreturnc                 ó  —  | j         |fi |¤Ž\  }}t          |                     ¦   «         ¦  «        D ]?}d|v r9d|v r|                     |¦  «        |d<   Œ#d|v r|                     |¦  «        |d<   Œ@ | j        |fi |¤ŽS )NÚnumÚheadsr   Úlayersr   )Úget_config_dictÚlistÚkeysÚpopÚ	from_dict)Úclsr!   r   Úconfig_dictÚks        r   Úfrom_pretrainedzMedusaConfig.from_pretrained&   sº   € ð 2˜cÔ1Ø)ð
ð 
Ø-3ð
ð 
ÑˆVõ k×&Ò&Ñ(Ô(Ñ)Ô)ð 	Jð 	JˆAØ˜ˆzˆzØ˜a<<Ø/:¯ª¸qÑ/AÔ/AK Ñ,Ð,Ø ]]Ø7B·²ÀqÑ7IÔ7IKÐ 3Ñ4øØˆsŒ}˜[Ð3Ð3¨FÐ3Ð3Ð3r    c                 ó   — dS )Nr   r   ©r   s    r   Únum_attention_headsz MedusaConfig.num_attention_heads7   s   € àˆqr    c                 ó   — | j         S ©N©r   r1   s    r   Únum_lookahead_tokensz!MedusaConfig.num_lookahead_tokens;   s
   € àŒ~Ðr    r6   c                 ó   — || _         d S r4   r5   )r   r6   s     r   r6   z!MedusaConfig.num_lookahead_tokens?   s   € à-ˆŒˆˆr    )r   r   r	   r
   r   r   N)Ú__name__Ú
__module__Ú__qualname__Ú
model_typer   r   ÚclassmethodÚstrÚosÚPathLiker/   Úpropertyr2   r6   ÚsetterÚ__classcell__)r   s   @r   r   r   	   s@  ø€ € € € € Ø€Jð  ØØØ!"ØØØ+/ð#ð #àð#ð ð#ð ð	#ð
 ð#ð ð#ð ð#ð " D™jð#ð #ð #ð #ð #ð #ð4 ð4à'*¨R¬[Ñ'8ð4ð 
ð	4ð 4ð 4ñ „[ð4ð  ðð ñ „Xðð ðð ñ „Xðð Ô ð.¸ð .ð .ð .ñ !Ô ð.ð .ð .ð .ð .r    r   )r>   Útransformersr   r   r   r    r   ú<module>rD      sW   ðð 
€	€	€	à )Ð )Ð )Ð )Ð )Ð )ð8.ð 8.ð 8.ð 8.ð 8.Ð#ñ 8.ô 8.ð 8.ð 8.ð 8.r    