§
    ‚Štjý  ã                   óŒ   — 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	d
S )é    )Ústricté   )ÚPreTrainedConfig)ÚRopeParameters)Úauto_docstringzCohereForAI/c4ai-command-r-v01)Ú
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	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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	dz  e
d0<   dZ$ee         dz  e
d1<   ˆ fd2„Z%ˆ xZ&S )3ÚCohere2Configa  
    logit_scale (`float`, *optional*, defaults to 0.0625):
        The scaling factor for the output logits.

    ```python
    >>> from transformers import Cohere2Model, Cohere2Config

    >>> # Initializing a Cohere Nextmodel configuration
    >>> configuration = Cohere2Config()

    >>> # Initializing a model from the Cohere2 configuration
    >>> model = Cohere2Model(configuration) # doctest: +SKIP

    >>> # Accessing the model configuration
    >>> configuration = model.config # doctest: +SKIP
    ```
    Úcohere2Úpast_key_valuesÚ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 X  Úintermediate_sizeg      °?Úlogit_scaleé(   Únum_hidden_layersé@   Únum_attention_headsNÚnum_key_value_headsÚsiluÚ
hidden_actÚmax_position_embeddingsg{®Gáz”?Úinitializer_rangegñhãˆµøä>Úlayer_norm_epsTÚ	use_cacher   Úpad_token_idé   Úbos_token_idiä Úeos_token_idÚtie_word_embeddingsÚrope_parametersFÚattention_biasg        Úattention_dropouti   Úsliding_windowÚlayer_typesc                 ó  •‡— | j         €| j        | _         | j        | j        z  | _        | j        €;|                     dd¦  «        Šˆfd„t          | j        ¦  «        D ¦   «         | _         t          ¦   «         j	        di |¤Ž d S )NÚsliding_window_patterné   c                 óB   •— g | ]}t          |d z   ‰z  ¦  «        rdnd‘ŒS )é   Úsliding_attentionÚfull_attention)Úbool)Ú.0ÚiÚ_sliding_window_patterns     €úo/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/transformers/models/cohere2/configuration_cohere2.pyú
<listcomp>z/Cohere2Config.__post_init__.<locals>.<listcomp>c   sG   ø€ ð  ð  ð  àõ (,¨Q°©UÐ6MÑ,MÑ'NÔ'NÐdÐ#Ð#ÐTdð ð  ð  ó    © )
r   r   r   Úhead_dimr.   ÚpopÚranger   ÚsuperÚ__post_init__)ÚselfÚkwargsr9   Ú	__class__s     @€r:   rB   zCohere2Config.__post_init__X   s§   øø€ ØÔ#Ð+Ø'+Ô'?ˆDÔ$ð Ô(¨DÔ,DÑDˆŒð ÔÐ#à&,§j¢jÐ1IÈ1Ñ&MÔ&MÐ#ð ð  ð  ð  å˜tÔ5Ñ6Ô6ð ñ  ô  ˆDÔð
 	�‰ŒÔÐ'Ð' Ð'Ð'Ð'Ð'Ð'r<   )'Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ú
model_typeÚkeys_to_ignore_at_inferenceÚbase_model_tp_planÚbase_model_pp_planr   ÚintÚ__annotations__r   r   r   Úfloatr   r   r   r    Ústrr!   r"   r#   r$   r6   r%   r'   r(   Úlistr)   r*   r   Údictr+   r,   r-   r.   rB   Ú__classcell__)rE   s   @r:   r
   r
      s?  ø€ € € € € € ðð ð$ €JØ#4Ð"5Ðà%.Ø%.Ø%.Ø%.Ø"+Ø )Ø"+ðð Ðð &˜¨Ð(9Ð:Ø#Ð%5Ð6¸Ð8IÐJØ!Ð" _Ð$5Ð6ðð Ðð €J�ÐÐÑØ€K�ÐÐÑØ"Ð�sÐ"Ð"Ñ"Ø€K�ÐÐÑØÐ�sÐÐÑØ!Ð˜Ð!Ð!Ñ!Ø&*Ð˜˜t™Ð*Ð*Ñ*Ø€J�ÐÐÑØ#'Ð˜SÐ'Ð'Ñ'Ø#Ð�uÐ#Ð#Ñ#Ø €N�EÐ Ð Ñ Ø€IˆtÐÐÑØ €L�#˜‘*Ð Ð Ñ Ø €L�#˜‘*Ð Ð Ñ Ø+1€L�#˜˜Sœ	‘/ DÑ(Ð1Ð1Ñ1Ø $Ð˜Ð$Ð$Ñ$Ø48€O�^ dÑ*¨TÑ1Ð8Ð8Ñ8Ø €N�DÐ Ð Ñ Ø%(Ð�u˜s‘{Ð(Ð(Ñ(Ø!%€N�C˜$‘JÐ%Ð%Ñ%Ø$(€K��c”˜TÑ!Ð(Ð(Ñ(ð(ð (ð (ð (ð (ð (ð (ð (ð (r<   r
   N)
Úhuggingface_hub.dataclassesr   Úconfiguration_utilsr   Úmodeling_rope_utilsr   Úutilsr   r
   Ú__all__r=   r<   r:   ú<module>rZ      s¶   ðð* /Ð .Ð .Ð .Ð .Ð .à 3Ð 3Ð 3Ð 3Ð 3Ð 3Ø 1Ð 1Ð 1Ð 1Ð 1Ð 1Ø #Ð #Ð #Ð #Ð #Ð #ð €Ð;Ð<Ñ<Ô<ØðJ(ð J(ð J(ð J(ð J(Ð$ñ J(ô J(ñ „ñ =Ô<ðJ(ðZ Ð
€€€r<   