§
    ‚Štj‹  ã                   ó˜   — d dl mZ 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_docstring)Úintervalzinceptionai/Jais-2-8B-Chat)Ú
checkpointc                   óö  ‡ — e Zd ZU dZdZdgZ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<     ed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  dz  e
d-<   d"Z"ee
d.<   dZ#e	dz  e
d/<   d0Z$ee
d1<   ˆ fd2„Z%d3„ Z&ˆ xZ'S )4ÚJais2Configat  
    ```python
    >>> from transformers import Jais2Model, Jais2Config

    >>> # Initializing a Jais2 jais2-7b style configuration
    >>> configuration = Jais2Config()

    >>> # Initializing a model from the jais2-7b style configuration
    >>> model = Jais2Model(configuration)

    >>> # Accessing the model configuration
    >>> configuration = model.config
    ```Újais2Ú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.up_projzlayers.*.mlp.down_projÚ	input_idsÚinputs_embedsÚhidden_statesÚattention_mask)Úembed_tokensÚlayersÚnormi K Ú
vocab_sizei   Úhidden_sizei h  Úintermediate_sizeé    Únum_hidden_layersé   Únum_attention_headsNÚnum_key_value_headsÚrelu2Ú
hidden_acti    Úmax_position_embeddingsg        g      ð?)ÚminÚmaxg{®Gáz”?)ÚdefaultÚinitializer_rangeTÚ	use_cacheÚpad_token_idr   Úbos_token_idiJ Úeos_token_idFÚtie_word_embeddingsÚrope_parametersÚattention_biasÚattention_dropoutÚmlp_biasÚhead_dimgñhãˆµøä>Úlayer_norm_epsc                 ó–   •— | j         €| j        | j        z  | _         | j        €| j        | _         t	          ¦   «         j        di |¤Ž d S )N© )r/   r   r   r   ÚsuperÚ__post_init__)ÚselfÚkwargsÚ	__class__s     €úk/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/transformers/models/jais2/configuration_jais2.pyr4   zJais2Config.__post_init__T   sT   ø€ ØŒ=Ð Ø Ô,°Ô0HÑHˆDŒMØÔ#Ð+Ø'+Ô'?ˆDÔ$à�‰ŒÔÐ'Ð' Ð'Ð'Ð'Ð'Ð'ó    c                 ól   — | j         | j        z  dk    r t          d| j         › d| j        › d�¦  «        ‚dS )zOPart of `@strict`-powered validation. Validates the architecture of the config.r   zThe hidden size (z6) is not a multiple of the number of attention heads (z).N)r   r   Ú
ValueError)r5   s    r8   Úvalidate_architecturez!Jais2Config.validate_architecture\   s[   € àÔ˜dÔ6Ñ6¸!Ò;Ð;Ýð7 DÔ$4ð 7ð 7ØÔ2ð7ð 7ð 7ñô ð ð <Ð;r9   )(Ú__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   r%   Úfloatr&   Úboolr'   r(   r)   Úlistr*   r+   r   Údictr,   r-   r.   r/   r0   r4   r<   Ú__classcell__)r7   s   @r8   r   r      sS  ø€ € € € € € ðð ð €JØ#4Ð"5Ðà%.Ø%.Ø%.Ø%.Ø )Ø"+ðð Ðð &˜¨Ð(9Ð:Ø#Ð%5Ð6¸Ð8IÐJØ!Ð" _Ð$5Ð6ðð Ðð €J�ÐÐÑØ€K�ÐÐÑØ"Ð�sÐ"Ð"Ñ"ØÐ�sÐÐÑØ!Ð˜Ð!Ð!Ñ!Ø&*Ð˜˜t™Ð*Ð*Ñ*Ø€J�ÐÐÑØ#'Ð˜SÐ'Ð'Ñ'Ø9˜x˜x¨C°SÐ9Ñ9Ô9À$ÐGÑGÔGÐ�uÐGÐGÑGØ€IˆtÐÐÑØ#€L�#˜‘*Ð#Ð#Ñ#Ø €L�#˜‘*Ð Ð Ñ Ø+1€L�#˜˜Sœ	‘/ DÑ(Ð1Ð1Ñ1Ø %Ð˜Ð%Ð%Ñ%Ø48€O�^ dÑ*¨TÑ1Ð8Ð8Ñ8Ø€N�DÐÐÑØ,/Ð�s˜U‘{ TÑ)Ð/Ð/Ñ/Ø€HˆdÐÐÑØ€Hˆc�D‰jÐÐÑØ €N�EÐ Ð Ñ ð(ð (ð (ð (ð (ðð ð ð ð ð ð r9   r   N)Úhuggingface_hub.dataclassesr   Úconfiguration_utilsr   Úmodeling_rope_utilsr   Úutilsr   Úutils.type_validatorsr   r   Ú__all__r2   r9   r8   ú<module>rS      sÇ   ðð, /Ð .Ð .Ð .Ð .Ð .à 3Ð 3Ð 3Ð 3Ð 3Ð 3Ø 1Ð 1Ð 1Ð 1Ð 1Ð 1Ø #Ð #Ð #Ð #Ð #Ð #Ø -Ð -Ð -Ð -Ð -Ð -ð €Ð7Ð8Ñ8Ô8ØðBð Bð Bð Bð BÐ"ñ Bô Bñ „ñ 9Ô8ðBðJ ˆ/€€€r9   