+
    ÒÜ³jL  ã                   ó8   € ^RI Ht ^ RIHtHt  ! R R]4      tR# )é   )ÚBaseAWQForCausalLM)ÚGPTBigCodeForCausalLMÚGPTBigCodeBlockc                   ó˜   a € ] tR t^t o RtRt]V 3R lR l4       t]V 3R lR l4       t]V 3R lR l4       t	]V 3R	 lR
 l4       t
RtV tR# )ÚGptBigCodeAWQForCausalLMr   Ún_positionsc                ó    <€ V ^8„  d   QhRS[ /# ©é   Úmodel©r   )ÚformatÚ__classdict__s   "€ÚW/home/luca/screen-checking/.venv/lib/python3.14/site-packages/awq/models/gpt_bigcode.pyÚ__annotate__Ú%GptBigCodeAWQForCausalLM.__annotate__   s   ø€ ÷ #ñ #Ñ 5ñ #ó    c                ó.   € V P                   P                  # ©N)ÚtransformerÚh)r   s   &r   Úget_model_layersÚ)GptBigCodeAWQForCausalLM.get_model_layers   s   € à× Ñ ×"Ñ"Ð"r   c                ó    <€ V ^8„  d   QhRS[ /# ©r   Úmodule©ÚOldGptBigCodeBlock)r   r   s   "€r   r   r      s   ø€ ÷ 
ñ 
Ñ$6ñ 
r   c                ó„   € \        R RV P                  P                  V P                  P                  P                  R7      # )Tzmlp.act)Úis_scalableÚ
scale_nameÚscale_layerÚscale_shape)ÚdictÚmlpÚactÚc_fcÚout_features)r   s   &r   Úget_act_for_scalingÚ,GptBigCodeAWQForCausalLM.get_act_for_scaling   s1   € äØØ ØŸ
™
Ÿ™ØŸ
™
Ÿ™×4Ñ4ô	
ð 	
r   c                ó    <€ V ^8„  d   QhRS[ /# r
   r   )r   r   s   "€r   r   r      s   ø€ ÷ Cñ CÑ/ñ Cr   c                ó>  € V P                   P                  P                  V4      V P                   n        V P                   P                  P                  V4      V P                   n        V P                   P                  P                  V4      V P                   n        R # r   )r   ÚwteÚtoÚwpeÚdrop)r   Údevices   &&r   Ú
move_embedÚ#GptBigCodeAWQForCausalLM.move_embed   sq   € à %× 1Ñ 1× 5Ñ 5× 8Ñ 8¸Ó @ˆ×ÑÔØ %× 1Ñ 1× 5Ñ 5× 8Ñ 8¸Ó @ˆ×ÑÔØ!&×!2Ñ!2×!7Ñ!7×!:Ñ!:¸6Ó!Bˆ×ÑÖr   c                ó    <€ V ^8„  d   QhRS[ /# r   r   )r   r   s   "€r   r   r       s   ø€ ÷ !ñ !Ñ'9ñ !r   c           
     óÞ  € . pVP                  \        V P                  V P                  P                  .VR ,          V P                  VR7      4       VP                  \        V P
                  V P                  P                  .VR,          V P                  R7      4       VP                  \        V P                  P                  V P                  P                  .VR,          R7      4       V# )zattn.c_attn)Úprev_opÚlayersÚinpÚmodule2inspectÚkwargszmlp.c_fc)r6   r7   r8   r9   z
mlp.c_proj)r6   r7   r8   )
Úappendr$   Úln_1ÚattnÚc_attnÚln_2r%   r'   r&   Úc_proj)r   Ú
input_featÚmodule_kwargsr7   s   &&& r   Úget_layers_for_scalingÚ/GptBigCodeAWQForCausalLM.get_layers_for_scaling   sÀ   € àˆð 	�‰ÜØŸ™ØŸ™×*Ñ*Ð+Ø˜}Õ-Ø%Ÿ{™{Ø$ôô	
ð 	�‰ÜØŸ™ØŸ
™
Ÿ™Ð(Ø˜zÕ*Ø%Ÿz™zô	ô	
ð 	�‰ÜØŸ
™
Ÿ™ØŸ
™
×)Ñ)Ð*Ø˜|Õ,ôô	
ð ˆr   © N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú
layer_typeÚmax_seq_len_keyÚstaticmethodr   r)   r2   rC   Ú__static_attributes__Ú__classdictcell__)r   s   @r   r   r      s_   ø‡ € Ø"€JØ#€Oà÷#ó ð#ð ÷
ó ð
ð ÷Có ðCð
 ÷!ó ö!r   r   N)Úbaser   Ú4transformers.models.gpt_bigcode.modeling_gpt_bigcoder   r   r   r   rE   r   r   Ú<module>rQ      s   ðÝ $÷ô9Ð1ö 9r   