+
    "Ý³j‚  ã                  ó¾   € ^ RI Ht ^ RIHtHtHtHtHt ^ RIt^ RI	H
t
 ^ RIt^ RIHt ^RIHt ^RIHt ^RIHt R R	 ltR
]/R R lltR R ltRRR
]/R R lltR# )é    )Úannotations)ÚAnyÚDictÚIterableÚOptionalÚTypeN)Ú_ConvNd)ÚNamedModule)ÚExllamaV3Linear)Úrecurse_setattrc               ó    € V ^8„  d   QhRRRR/# )é   Ú	submoduleú	nn.ModuleÚreturnztuple[int, int]© )Úformats   "ÚZ/home/luca/screen-checking/.venv/lib/python3.14/site-packages/gptqmodel/utils/exllamav3.pyÚ__annotate__r      s   € ÷ \ñ \ Yð \°?ñ \ó    c                ó>  € \        V \        4      '       d   T MR pVe   VP                  MT pVeU   VP                  P	                  R4      pVP                  P	                  R4      pVe   Ve   \        V4      \        V4      3# \        V\        P                  4      '       d   VP                  VP                  3# \        V\        4      '       d   VP                  VP                  3# \        V\        P                  4      '       d;   VP                  P                   ^ ,          VP                  P                   ^,          3# \#        VRR 4      p\#        VRR 4      pVe   Ve   \        V4      \        V4      3# \%        RVP&                  P(                   24      h)NÚin_featuresÚout_featureszUnsupported EXL3 module type: )Ú
isinstancer
   ÚmoduleÚstateÚgetÚintÚnnÚLinearr   r   r	   Úin_channelsÚout_channelsÚtransformersÚConv1DÚweightÚshapeÚgetattrÚNotImplementedErrorÚ	__class__Ú__name__)r   ÚnamedÚtargetr   r   s   &    r   Ú_resolve_linear_shaper-      sS  € Ü# I¬{×;Ò;‰IÀ€EØ"Ò.ˆU�\Š\°I€FàÒØ—k‘k—o‘o mÓ4ˆØ—{‘{—‘ ~Ó6ˆØÒ" |Ò'?Ü�{Ó#¤S¨Ó%6Ð6Ð6ä�&œ"Ÿ)™)×$Ò$Ø×!Ñ! 6×#6Ñ#6Ð6Ð6Ü�&œ'×"Ò"Ø×!Ñ! 6×#6Ñ#6Ð6Ð6Ü�&œ,×-Ñ-×.Ò.Ø�}‰}×"Ñ" 1Õ% v§}¡}×':Ñ':¸1Õ'=Ð=Ð=ä˜& -°Ó6€KÜ˜6 >°4Ó8€LØÒ <Ò#;Ü�;Ó¤ \Ó!2Ð2Ð2ä
Ð >¸v×?OÑ?O×?XÑ?XÐ>YÐZÓ
[Ð[r   Ú
module_clsc               ó0   € V ^8„  d   QhRRRRRRRRRR	R
R/# )r   Úmodule_rootr   ÚnameÚstrr   ÚtensorszDict[str, torch.Tensor]r.   úType[nn.Module]r   r   )r   s   "r   r   r   0   sF   € ÷ ñ àðð ðð ð	ð
 %ðð  ðð ñr   c                ób   € \        V4      w  rVVP                  VVVVR 7      p\        WV4       V# ))r   r   r1   r3   )r-   Úfrom_tensorsr   )r0   r1   r   r3   r.   r   r   Ú
new_modules   $$$$$   r   Úcreate_exllamav3_moduler8   0   sD   € ô !6°iÓ @Ñ€KØ×(Ñ(ØØ!ØØð	 )ó €Jô �K zÔ2ØÐr   c               ó    € V ^8„  d   QhRRRR/# )r   Úmodelr   r   zDict[str, Dict[str, Any]]r   )r   s   "r   r   r   C   s   € ÷ ñ ¨)ð Ð8Qñ r   c                ó°   € / pV P                  4        F?  w  r#\        VR R4      R8X  g   K  \        VR4      '       g   K-  VP                  4       W&   KA  	  V# )Ú
QUANT_TYPENÚexl3Útensor_storage_entry)Únamed_modulesr'   Úhasattrr>   )r:   Ústorager1   r   s   &   r   Úbuild_exllamav3_tensor_storagerB   C   sQ   € Ø)+€GØ×+Ñ+Ö-‰ˆÜ�6˜<¨Ó.°&Ö8¼WÀVÐMc×=dÔ=dØ"×7Ñ7Ó9ˆG‹Mñ .ð €Nr   Útensor_storagec          
     ó,   € V ^8„  d   QhRRRRRRRRR	R
/# )r   r:   r   Úmodule_nameszIterable[str]rC   z#Optional[Dict[str, Dict[str, Any]]]r.   r4   r   ÚNoner   )r   s   "r   r   r   K   s<   € ÷ 8ñ 8àð8ð  ð8ð 8ð	8ð
  ð8ð 
ñ8r   c           	     óf  € \        V P                  4       4      pT;'       g    / pV F…  pVP                  V4      pVf   K  \        V\        P
                  \        P                  \        34      '       g   KQ  \        V4      w  r‰V! VV	VVP                  V4      R7      p
\        WV
4       K‡  	  R # )N)r   r   r1   rC   )Údictr?   r   r   r   r    r#   r$   r	   r-   r   )r:   rE   rC   r.   Úmodule_lookupÚstorage_mapÚmodule_namer   r   r   r7   s   $$$$       r   Úreplace_exllamav3_placeholdersrL   K   s    € ô ˜×,Ñ,Ó.Ó/€MØ ×&Ð& B€Kã#ˆØ!×%Ñ% kÓ2ˆ	ØÒÙä˜)¤b§i¡i´×1DÑ1DÄgÐ%N×OÒOÙä$9¸)Ó$DÑ!ˆÙØ#Ø%ØØ&Ÿ?™?¨;Ó7ô	
ˆ
ô 	˜¨JÖ7ó $r   )Ú
__future__r   Útypingr   r   r   r   r   ÚtorchÚtorch.nnr   r#   Útorch.nn.modules.convr	   Úlooper.named_moduler
   Únn_modules.exllamav3r   r:   r   r-   r8   rB   rL   r   r   r   Ú<module>rT      s]   ðõ #ç 6Õ 6ã Ý Û Ý )å -Ý 2Ý "õ\ð2ð #2÷õ&ð8ð ;?ð	8ð
 #2÷8ñ 8r   