+
    œ°³j‹  ã                   ó€   € ^ RI Ht ^RIHt ^RIHt ^RIHt ^RIH	t	 ]! RR7      ] ! R R	]4      4       4       t
R	.tR
# )é    )Ústrict)ÚPreTrainedConfig)ÚRopeParameters)Úauto_docstring)Úintervalzinceptionai/Jais-2-8B-Chat)Ú
checkpointc                   ó  a a€ ] tR t^t oRtRtR.tRRRRRRRR	R
RRR	/tRR.R.3RRR.R.3RR.R.3/tRt	Rt
Rt^ t^tRtRtRt]! RRR7      ! RR7      tRtRt^ tRtR tRtRtRtRtRtR!tV 3R" ltR# tV3R$ lt R%t!Vt"V ;t## )&ÚJais2ConfigaP  
```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_valueszlayers.*.self_attn.q_projÚcolwisezlayers.*.self_attn.k_projzlayers.*.self_attn.v_projzlayers.*.self_attn.o_projÚrowwisezlayers.*.mlp.up_projzlayers.*.mlp.down_projÚembed_tokensÚ	input_idsÚinputs_embedsÚlayersÚhidden_statesÚattention_maskÚnormi K i   i h  NÚrelu2i    g        g      ð?)ÚminÚmaxg{®Gáz”?)ÚdefaultTiJ Fgñhãˆµøä>c                óÆ   <€ V P                   f#   V P                  V P                  ,          V n         V P                  f   V P                  V n        \        SV `  ! R/ VB  R # )N© )Úhead_dimÚhidden_sizeÚnum_attention_headsÚnum_key_value_headsÚsuperÚ__post_init__)ÚselfÚkwargsÚ	__class__s   &,€Ún/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/models/jais2/configuration_jais2.pyr!   ÚJais2Config.__post_init__T   sO   ø€ Ø�=‰=Ò Ø ×,Ñ,°×0HÑ0HÕHˆDŒMØ×#Ñ#Ò+Ø'+×'?Ñ'?ˆDÔ$ä‰ÒÑ' Ô'ó    c                ó–   € V P                   V P                  ,          ^ 8w  d'   \        RV P                    RV P                   R24      hR# )zOPart of `@strict`-powered validation. Validates the architecture of the config.zThe hidden size (z6) is not a multiple of the number of attention heads (z).N)r   r   Ú
ValueError)r"   s   &r%   Úvalidate_architectureÚ!Jais2Config.validate_architecture\   sS   € à×Ñ˜d×6Ñ6Õ6¸!Ô;ÜØ# D×$4Ñ$4Ð#5ð 6Ø×2Ñ2Ð3°2ð7óð ñ <r'   c                ó¬  <€ V ^8„  d   Qh/ S[ ;R&   S[ ;R&   S[ ;R&   S[ ;R&   S[ ;R&   S[ R,          ;R&   S[;R&   S[ ;R	&   S[;R
&   S[;R&   S[ R,          ;R&   S[ R,          ;R&   S[ S[S[ ,          ,          R,          ;R&   S[;R&   S[S[,          R,          ;R&   S[;R&   S[ S[,          R,          ;R&   S[;R&   S[ R,          ;R&   S[;R&   # )é   Ú
vocab_sizer   Úintermediate_sizeÚnum_hidden_layersr   Nr   Ú
hidden_actÚmax_position_embeddingsÚinitializer_rangeÚ	use_cacheÚpad_token_idÚbos_token_idÚeos_token_idÚtie_word_embeddingsÚrope_parametersÚattention_biasÚattention_dropoutÚmlp_biasr   Úlayer_norm_eps)ÚintÚstrÚfloatÚboolÚlistr   Údict)ÚformatÚ__classdict__s   "€r%   Ú__annotate__ÚJais2Config.__annotate__   sa  ø‡ ‚ ñB ÑñC ñD ÑñE ñF Ñ"ñG ñH ÑñI ñJ Ñ!ñK ñL ˜t�Ñ*ñM ñN ÑñO ñP !Ñ'ñQ ñR ÑGñS ñT ÑñU ñV ˜•*Ñ#ñW ñX ˜•*Ñ ñY ñZ ™™S�	•/ DÕ(Ñ1ñ[ ñ\ Ñ%ñ] ñ^ $¡dÕ*¨TÕ1Ñ8ñ_ ñ` Ñña ñb ™U•{ TÕ)Ñ/ñc ñd Ññe ñf �D�jÑñg ñh Ñ òi r'   )r   r   )$Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__Ú
model_typeÚkeys_to_ignore_at_inferenceÚbase_model_tp_planÚbase_model_pp_planr.   r   r/   r0   r   r   r1   r2   r   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r   r=   r!   r*   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__Ú__classcell__)r$   rE   s   @@r%   r
   r
      s  ù‡ € ñð €JØ#4Ð"5Ðà# YØ# YØ# YØ# YØ 	Ø  )ðÐð 	˜+˜¨Ð(9Ð:Ø�OÐ%5Ð6¸Ð8IÐJØ�/Ð" _Ð$5Ð6ðÐð €JØ€KØ"ÐØÐØ!ÐØ&*ÐØ€JØ#'ÐÙ'¨C°SÕ9À$ÔGÐØ€IØ#€LØ €LØ+1€LØ %ÐØ48€OØ€NØ,/ÐØ€HØ€HØ €Nõ(ò÷} … r'   r
   N)Úhuggingface_hub.dataclassesr   Úconfiguration_utilsr   Úmodeling_rope_utilsr   Úutilsr   Úutils.type_validatorsr   r
   Ú__all__r   r'   r%   Ú<module>r[      sM   ðõ, /å 3Ý 1Ý #Ý -ñ Ð7Ô8ØôBÐ"ó Bó ó 9ðBðJ ˆ/‚r'   