Ë
    ÜÓ^j�  ã                   ór   — d dl mZ ddlmZ ddlmZ ddlmZ  ed¬«      e G d„ d	e«      «       «       Zd	gZ	y
)é    )Ústricté   )ÚPreTrainedConfig)ÚRopeParameters)Úauto_docstringzallenai/Olmo2-7B-1124-hf)Ú
checkpointc                   óš  ‡ — e Zd ZU dZdZdgZddddddddœZd	gd
gfddgdgfdgdgfdœZdZe	e
d<   dZe	e
d<   dZe	e
d<   dZe	e
d<   dZe	e
d<   dZe	dz  e
d<   dZee
d<   dZe	e
d<   dZee
d<   dZee
d <   d!Ze	dz  e
d"<   dZe	dz  e
d#<   d$Ze	ee	   z  dz  e
d%<   d&Zee
d'<   dZeez  dz  e
d(<   d&Zee
d)<   d*Z ee	z  e
d+<   d,Z!ee
d-<   ˆ fd.„Z"ˆ xZ#S )/ÚOlmo2Configa�  
    Example:

    ```python
    >>> from transformers import Olmo2Model, Olmo2Config

    >>> # Initializing a Olmo2 7B style configuration
    >>> configuration = Olmo2Config()

    >>> # Initializing a model from the Olmo2 7B style configuration
    >>> model = Olmo2Model(configuration)

    >>> # Accessing the model configuration
    >>> configuration = model.config
    ```
    Úolmo2Úpast_key_valuesÚcolwise_gather_outputÚrowwise_split_inputÚcolwiseÚrowwise)zlayers.*.self_attn.q_projzlayers.*.self_attn.k_projzlayers.*.self_attn.v_projzlayers.*.self_attn.o_projzlayers.*.mlp.gate_projzlayers.*.mlp.up_projzlayers.*.mlp.down_projÚ	input_idsÚinputs_embedsÚhidden_statesÚattention_mask)Úembed_tokensÚlayersÚnormi€Ä  Ú
vocab_sizei   Úhidden_sizei +  Úintermediate_sizeé    Únum_hidden_layersÚnum_attention_headsNÚnum_key_value_headsÚsiluÚ
hidden_acti   Úmax_position_embeddingsg{®Gáz”?Úinitializer_rangeTÚ	use_cacheé   Úpad_token_idÚbos_token_idigÄ  Úeos_token_idFÚtie_word_embeddingsÚrope_parametersÚattention_biasg        Úattention_dropoutgñhãˆµøä>Úrms_norm_epsc                 ó^   •— | j                   €| j                  | _         t        ‰| �  di |¤Ž y )N© )r   r   ÚsuperÚ__post_init__)ÚselfÚkwargsÚ	__class__s     €úx/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/transformers/models/olmo2/configuration_olmo2.pyr0   zOlmo2Config.__post_init__Z   s-   ø€ Ø×#Ñ#Ð+Ø'+×'?Ñ'?ˆDÔ$Ü‰ÑÑ' Ó'ó    )$Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ú
model_typeÚkeys_to_ignore_at_inferenceÚbase_model_tp_planÚbase_model_pp_planr   ÚintÚ__annotations__r   r   r   r   r   r    Ústrr!   r"   Úfloatr#   Úboolr%   r&   r'   Úlistr(   r)   r   Údictr*   r+   r,   r0   Ú__classcell__)r3   s   @r4   r
   r
   !   sZ  ø… ñð" €JØ#4Ð"5Ðà%<Ø%<Ø%<Ø%:Ø"+Ø )Ø"+ñÐð &˜¨Ð(9Ð:Ø#Ð%5Ð6¸Ð8IÐJØ!Ð" _Ð$5Ð6ñÐð €J�ÓØ€K�ÓØ"Ð�sÓ"ØÐ�sÓØ!Ð˜Ó!Ø&*Ð˜˜t™Ó*Ø€J�ÓØ#'Ð˜SÓ'Ø#Ð�uÓ#Ø€IˆtÓØ €L�#˜‘*Ó Ø#€L�#˜‘*Ó#Ø+0€L�#˜˜S™	‘/ DÑ(Ó0Ø %Ð˜Ó%Ø48€O�^ dÑ*¨TÑ1Ó8Ø €N�DÓ Ø%(Ð�u˜s‘{Ó(à€L�%Ó÷(ð (r5   r
   N)
Úhuggingface_hub.dataclassesr   Úconfiguration_utilsr   Úmodeling_rope_utilsr   Úutilsr   r
   Ú__all__r.   r5   r4   ú<module>rK      sG   ðõ4 /å 3Ý 1Ý #ñ Ð5Ô6Øô:(Ð"ó :(ó ó 7ð:(ðz ˆ/�r5   