
    3jp                        S SK Jr  S SKrS SKrS SKrS SKrS SKJr  S SKJ	r	J
r
Jr  S SKJr  SSKJr  S SKr\R"                  " S5      r\ " S	 S
5      5       rg)    )annotationsN)Path)AnyLiteralOptional)	dataclass   )Keysmetadatac                     \ rS rSr% SrS\S'   SrS\S'   SrS\S'   SrS\S	'   Sr	S\S
'   Sr
S\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S'   SrS\S '   Sr S\S!'   Sr!S\S"'   Sr"S\S#'   Sr#S\S$'   Sr$S\S%'   Sr%S&\S''   Sr&S(\S)'   Sr'S(\S*'   Sr(S&\S+'   \)S6S7S, jj5       r*\)S8S9S- jj5       r+\)S8S:S. jj5       r,\)S8S:S/ jj5       r-\)S8S:S0 jj5       r.\)S1 5       r/\)S;S<S2 jj5       r0\)S6S=S3 jj5       r1S>S4 jr2S5r3g)?Metadata   NOptional[str]sampling_sequencezOptional[int]sampling_top_kzOptional[float]sampling_top_psampling_min_psampling_xtc_probabilitysampling_xtc_thresholdsampling_tempsampling_penalty_last_nsampling_penalty_repeatsampling_mirostatsampling_mirostat_tausampling_mirostat_etanameauthorversionorganizationfinetunebasenamedescriptionquantized_by
size_labelurldoiuuidrepo_url
source_url
source_doisource_uuidsource_repo_urllicenselicense_namelicense_linkzOptional[list[dict]]base_modelszOptional[list[str]]tags	languagesdatasetsc                `   [        5       n[         R                  U5      n[         R                  U5      n[         R                  U5      n[         R	                  XEXaU5      nU(       Ga  UR                  SUR                  5      Ul        UR                  SUR                  5      Ul        UR                  SUR                  5      Ul        UR                  SUR                  5      Ul	        UR                  SUR                  5      Ul
        UR                  SUR                  5      Ul        UR                  SUR                  5      Ul        UR                  SUR                  5      Ul        UR                  S	UR                  5      Ul        UR                  S
UR                  5      Ul        UR                  SUR                   5      Ul        UR                  SUR"                  5      Ul        [         R%                  U 5      nUR                  [&        R(                  R*                  UR                  5      Ul        UR                  [&        R(                  R,                  UR                  5      Ul        UR                  [&        R(                  R.                  UR                  5      Ul        UR                  [&        R(                  R0                  UR                  5      Ul	        UR                  [&        R(                  R2                  UR                  5      Ul
        UR                  [&        R(                  R4                  UR                  5      Ul        UR                  [&        R(                  R6                  UR                  5      Ul        UR                  [&        R(                  R8                  UR                  5      Ul        UR                  [&        R(                  R:                  UR                  5      Ul        UR                  [&        R(                  R<                  UR                  5      Ul        UR                  [&        R(                  R>                  UR                   5      Ul        UR                  [&        R(                  R@                  UR"                  5      Ul        UR                  [&        R(                  RB                  URD                  5      Ul"        UR                  [&        R(                  RF                  URH                  5      Ul$        UR                  [&        R(                  RJ                  URL                  5      Ul&        UR                  [&        R(                  RN                  URP                  5      Ul(        UR                  [&        R(                  RR                  URT                  5      Ul*        UR                  [&        R(                  RV                  URX                  5      Ul,        UR                  [&        R(                  RZ                  UR\                  5      Ul.        UR                  [&        R(                  R^                  UR`                  5      Ul0        UR                  [&        R(                  Rb                  URd                  5      Ul2        UR                  [&        R(                  Rf                  URh                  5      Ul4        UR                  [&        R(                  Rj                  URl                  5      Ul6        UR                  [&        R(                  Rn                  URp                  5      Ul8        UR                  [&        R(                  Rr                  URt                  5      Ul:        UR                  [&        R(                  Rv                  URx                  5      Ul<        UR                  [&        R(                  Rz                  UR|                  5      Ul>        UR                  [&        R(                  R~                  UR                  5      Ul@        UR                  [&        R(                  R                  UR                  5      UlB        UR                  [&        R(                  R                  UR                  5      UlD        UR                  [&        R(                  R                  UR                  5      UlF        UR                  SUR                  5      UlG        UR                  SUR                  5      UlH        UR                  [&        R(                  R                  UR                  5      UlJ        UR                  [&        R(                  R                  UR                  5      UlL        Ub  X$l"        U$ )Nsequencetop_ktop_pmin_pxtc_probabilityxtc_thresholdtemperaturepenalty_last_npenalty_repeatmirostatmirostat_taumirostat_etazgeneral.base_modelszgeneral.datasets)Mr   load_model_cardload_hf_parametersload_generation_configapply_metadata_heuristicgetr   r   r   r   r   r   r   r   r   r   r   r   load_metadata_overrider
   GeneralSAMPLING_SEQUENCESAMPLING_TOP_KSAMPLING_TOP_PSAMPLING_MIN_PSAMPLING_XTC_PROBABILITYSAMPLING_XTC_THRESHOLDSAMPLING_TEMPSAMPLING_PENALTY_LAST_NSAMPLING_PENALTY_REPEATSAMPLING_MIROSTATSAMPLING_MIROSTAT_TAUSAMPLING_MIROSTAT_ETANAMEr   AUTHORr   VERSIONr   ORGANIZATIONr   FINETUNEr    BASENAMEr!   DESCRIPTIONr"   QUANTIZED_BYr#   
SIZE_LABELr$   LICENSE_NAMEr.   LICENSE_LINKr/   URLr%   DOIr&   UUIDr'   REPO_URLr(   
SOURCE_URLr)   
SOURCE_DOIr*   SOURCE_UUIDr+   SOURCE_REPO_URLr,   r0   r3   TAGSr1   	LANGUAGESr2   )	metadata_override_path
model_path
model_nametotal_paramsr   
model_card	hf_params
gen_configmetadata_overrides	            G/home/wildlama/miniconda3/lib/python3.13/site-packages/gguf/metadata.pyloadMetadata.load<   sP    :--j9
//
;	44Z@
 44X9bno0:zRZRlRl0mH&0:wRZRiRi0jH#0:wRZRiRi0jH#0:wRZRiRi0jH#0:?PRZRsRs0tH-0:RZRqRq0rH+0:}RZRhRh0iH"0:?ORZRrRr0sH,0:?ORZRrRr0sH,0:zRZRlRl0mH&0:~RZRpRp0qH*0:~RZRpRp0qH* %;;<RS,=,A,A$,,B`B`iq  jD  jD  -E",=,A,A$,,B]B]iq  jA  jA  -B,=,A,A$,,B]B]iq  jA  jA  -B,=,A,A$,,B]B]iq  jA  jA  -B,=,A,A$,,BgBgiq  jK  jK  -L),=,A,A$,,BeBeiq  jI  jI  -J',=,A,A$,,B\B\iqii  -A,=,A,A$,,BfBfiq  jJ  jJ  -K(,=,A,A$,,BfBfiq  jJ  jJ  -K(,=,A,A$,,B`B`iq  jD  jD  -E",=,A,A$,,BdBdiq  jH  jH  -I&,=,A,A$,,BdBdiq  jH  jH  -I&#4#8#89J9JW_WdWd#e#4#8#89L9LW_WfWf#g#4#8#89M9MW_WgWg#h#4#8#89R9RW_WlWl#m#4#8#89N9NW_WhWh#i#4#8#89N9NW_WhWh#i#4#8#89Q9QW_WkWk#l#4#8#89R9RW_WlWl#m#4#8#89P9PW_WjWj#k#4#8#89R9RW_WlWl#m#4#8#89R9RW_WlWl#m#4#8#89I9IW_WcWc#d#4#8#89I9IW_WcWc#d#4#8#89J9JW_WdWd#e#4#8#89N9NW_WhWh#i#4#8#89P9PW_WjWj#k#4#8#89P9PW_WjWj#k#4#8#89Q9QW_WkWk#l#4#8#89U9UW_WoWo#p  $5#8#89NW_WkWk#l $5#8#89KW_WhWh#i#4#8#89J9JW_WdWd#e#4#8#89O9OW_WiWi#j !&M    c                    U b  U R                  5       (       d  0 $ [        U SSS9 n[        R                  " U5      sS S S 5        $ ! , (       d  f       g = f)Nrutf-8encoding)is_fileopenjsonrr   )ri   fs     rq   rF   Metadata.load_metadata_override   sC    !)1G1O1O1Q1QI(#@A99Q< A@@s   A
Ac                   U b  U R                  5       (       d  0 $ U S-  nUR                  5       (       d  0 $ Sn[        USSS9 nUR                  5       nUR	                  5       n/ n[        U5      S:X  a  0 sS S S 5        $ [        U5      S:  a  US   S:w  a  0 sS S S 5        $ USS   H  nUS:X  a    OUR                  U5        M     S	R                  U5      S	-   nS S S 5        UR                  S
S5      nUR                  SS5      nU(       aQ  [        R                  " U5      n[        U[        5      (       a  U$ [        R                  S[        U5       S35        0 $ 0 $ ! , (       d  f       N= f)Nz	README.md rv   rw   rx   r   z---r	   
z- no
z- "no"
	z  z3while reading YAML model card frontmatter, data is z instead of dict)is_dirrz   r{   read
splitlineslenappendjoinreplaceyaml	safe_load
isinstancedictloggererrortype)	rj   model_card_pathyaml_contentr}   contentlines
lines_yamllinedatas	            rq   rA   Metadata.load_model_card   sg   Z%6%6%8%8I${2&&((I /39QffhG&&(EJ5zQ :9 5zA~%(e"3 :9 ab	5=%%d+	 "
  99Z047L :& $++HlC $++D$7>>,/D$%%RSWX\S]R^^nop	IC :9s   3E#E#$9E##
E1c                    U b  U R                  5       (       d  0 $ U S-  nUR                  5       (       d  0 $ [        USSS9 n[        R                  " U5      sS S S 5        $ ! , (       d  f       g = f)Nzconfig.jsonrv   rw   rx   )r   rz   r{   r|   rr   )rj   config_pathr}   s      rq   rB   Metadata.load_hf_parameters   s^    Z%6%6%8%8I =0""$$I+sW599Q< 655s   A""
A0c                ,   U b  U R                  5       (       d  0 $ U S-  nUR                  5       (       d  0 $  [        USSS9 n[        R                  " U5      sS S S 5        $ ! , (       d  f       g = f! [        R
                  [        4 a    0 s $ f = f)Nzgeneration_config.jsonrv   rw   rx   )r   rz   r{   r|   rr   JSONDecodeErrorIOError)rj   generation_config_pathr}   s      rq   rC   Metadata.load_generation_config   s    Z%6%6%8%8I!+.F!F%--//I	,cGDyy| EDD$$g. 	I	s/   A4 A#	A4 #
A1-A4 1A4 4BBc           	     (   SR                  U R                  5       R                  SS5      R                  5        Vs/ s HE  oR	                  5       (       a,  [
        R                  " SU5      (       d  UR                  5       OUPMG     sn5      $ s  snf )N -z^(v\d+(?:\.\d+)*|\d.*)$)r   stripr   splitislowerrematchtitle)stringws     rq   id_to_titleMetadata.id_to_title   s     xxrxr~r~  sA  sI  sI  JM  OR  sS  sY  sY  s[  \  s[mniikk"((C]_`:a:aghh  s[  \  ]  	]  \s   ABc           	     
   U c  gSU ;   a  U S S S S S 4$ SU ;   a  U R                  SS5      u  p#OS U p2Ub  [        U5      S:  a  US   S:X  a  S nUR                  S5      n[        [        [        U5      5      5       H  n[        XE   5      S:X  d  M  XE	 M     U Vs/ s H  n[	        5       PM     snn[        U5       GH  u  pX[        R                  " SU[        R                  5      (       a  Xu   R                  S	5        MF  [        R                  " S
U[        R                  5      (       a'  Xu   R                  S5        UR                  5       XE'   M  US:  Gas  [        R                  " SU[        R                  5      (       GaG  UR                  SS5      nUS   R                  5       (       a  US S S-   US   -   US   -   n[        U5      S:  a:  US   R                  5       (       a"  US   S;   a  US S US   R                  5       -   nUS:w  a   [        US S 5      [        SSR                  US   5      5      -  n	US:  a  U	[!        U5      S-  :  d  US:  aC  [!        X-
  5      SU-  S-  :  a,  Xu   R                  S5        US S US   R#                  5       -   n[        Xu   5      S:X  a  Xu   R                  S5        XU'   GM  US:  d  GM  [        R                  " SU[        R                  5      (       d  GMI  US:  a*  UR#                  5       S:X  a  Xu   R                  S5        GMy  Xu   R                  S5        GM     ['        S [)        XG5       5       5      (       aF  [)        XG5       H7  u  pSU;   d  M  [+        S U
 5       5      (       d  M&  UR-                  S5        M9     Sn[)        XG5       Hw  u  pU(       a@  [        U5      S:X  a  US   R/                  5       (       d  S	U;   a  UR                  S5        ML  U(       a  Sn[        U5      S:X  d  Mf  UR                  S5        My     [)        [        U5      [        U5      5       H-  u  pSU;   a"  [        U5      S:  a  UR-                  S5        M-    O   SR1                  S [)        XG5       5       5      =(       d    S nSR1                  [2        R5                  S [)        XG5       5       5      R7                  5       5      =(       d    S nSR1                  S  [)        XG5       5       5      =(       d    S nSR1                  S! [)        XG5       5       5      =(       d    S nUc  Uc  Uc  S nX2XUU4$ s  snf ! [$         a     GNf = f)"N)NNNNNNr   /r	   r   .r   z(v|iter)?\d+([.]\d+)*r   zi?q\d(_\w)*|b?fp?(16|32)r   zD(([A]|\d+[x])?\d+([._]\d+)?[KMBT][\d]?|small|mini|medium|large|x?xl)_kmbti  z KMBT      r    r$   zchat|instruct|vision|loralorac              3  h   #    U  H(  u  pS U;   d  M  U  H  o3R                  5       v   M     M*     g7fr$   N)	isdecimal).0ntcs       rq   	<genexpr>3Metadata.get_model_id_components.<locals>.<genexpr>F  s/     e)DXYHY}cd^_{{}}cd})Ds   2 2c              3  @   #    U  H  oR                  5       v   M     g 7fN)isalpha)r   r   s     rq   r   r   I  s     2199;;s   Tr!   Fc              3  :   #    U  H  u  pS U;   d  M  Uv   M     g7f)r!   N )r   r   r   s      rq   r   r   ^       Y*E$!WXAA*E   	c              3  :   #    U  H  u  pS U;   d  M  Uv   M     g7fr   r   )r   sr   s      rq   r   r   `  s     +k:U$!YeijYjAA:Ur   c              3  :   #    U  H  u  pS U;   d  M  Uv   M     g7f)r    Nr   )r   r}   r   s      rq   r   r   a  r   r   c              3  J   #    U  H  u  pS U;   d  M  SU;  d  M  Uv   M     g7f)r   r!   Nr   )r   vr   s      rq   r   r   d  s(     p*EVW1\fno\o11*Es   ##	#)r   r   reversedrangeset	enumerater   	fullmatch
IGNORECASEaddupperr   r   floatpowfindabslower
ValueErroranyzipallremover   r   r   fromkeyskeys)model_idrl   org_componentmodel_full_name_component
name_partsir   
name_typespartlabel_paramsr   r   at_startr!   r$   r    r   s                    rq   get_model_id_components Metadata.get_model_id_components   s   
 5(? T4tT99(?7?~~c17M4M4 8<X4 $]);a)?MRSDTX[D[ M 9 ? ? D
 %J01A:=!Q&M 2 ''JqSUJ' 	
 !,GA||4dBMMJJ!!),94OO!!&) $


Q2<<(oquwy  xE  xE  F  F||C-8%%''9s?T"X5R@Dt9q=T"X%7%7%9%9Bx6)#CRy48>>+;;1$',T#2Y'7#dGLLQUVXQYDZ:['[
 )1,L@QUV@V1V(1,\5P1QTUXdTdhiTi1i 'M--j9#'9tBx~~/?#?D z}%*M%%l3 $1Q2<<(DdBMMZZ!#

(>M%%f-M%%j1[ -b eZ)DeeeJ31$2222. 4
 :2GDc!fkd1goo.?.?IQRNj!$Hq6Q;EE*% 3 8J/*1EFGDQ3q6A:$	 G 88Y#j*EYYa]aXXdmm+k#j:U+kkpprs{w{
88Y#j*EYYa]a ((p#j*Eppxtx("2wH(WV```q (H & s   U&BU++
U98U9c           
     z  ^ ^ TGb  S;U U4S jjnS;U U4S jjnU" SS5        U" SS5        U" SS5        U" SS5        U" SS5        U" SS5        U" S	S	5        U" S
S
5        U" SS5        U" SS5        U" SS5        U" SS5        U" SS5        U" SS5        U" SS5        U" SS5        U" SS5        U" SS5        U" S	S5        U" S
S5        U" SS5        U" SS5        U" SS5        U" SS5        U" SS5        U" SS5        U" S	S 5        S!T;   d  S"T;   d  S#T;   GaP  / nTR                  S!TR                  S"TR                  S#S 5      5      5      nUbM  [        U[        5      (       a  UR                  U5        O&[        U[        5      (       a  UR                  U5        T R                  c  / T l        U GH  n	0 n
[        U	[        5      (       Ga@  U	R                  S$5      (       d,  U	R                  S%5      (       d  U	R                  S&5      (       a  XS'   S'U	;   a  [        R                  " S(U	5      nU(       aj  UR                  S)5      n[        R                  X5      u  pnnnnUb  [        R                  U5      U
S'   Ub  [        R                  U5      U
S'   Ub  UU
S'   O[        R                  X5      u  pnnnnUb  [        R                  U5      U
S'   Ub  [        R                  U5      U
S'   Ub  UU
S'   Ub  Ub  S*U S+U 3U
S'   O:[        U	[        5      (       a  U	n
O"[        R!                  S,[        U	5       S-35        T R                  R                  U
5        GM     S.T;   d  S/T;   d  S0T;   Ga'  / nTR                  S.TR                  S/TR                  S0S 5      5      5      nUbM  [        U[        5      (       a  UR                  U5        O&[        U[        5      (       a  UR                  U5        T R"                  c  / T l        U GH  n0 n[        U[        5      (       Ga  UR                  S15      (       a  UUS'   S'U;   a  [        R                  " S(U5      nU(       ak  UR                  S)5      n[        R                  UU5      u  npnnnUb  [        R                  U5      US'   Ub  [        R                  U5      US'   Ub  UUS'   O[        R                  UU5      u  npnnnUb  [        R                  U5      US'   Ub  [        R                  U5      US'   Ub  UUS'   Ub  Ub  S*U S+U 3US'   O:[        U[        5      (       a  UnO"[        R!                  S2[        U5       S-35        T R"                  R                  U5        GM     U" S3S35        U" S4S45        U" S5S55        U" S6S65        U" S6S75        U" S8S85        U" S8S95        Ub  UR                  S:5      nUb  UR%                  S+5      S)::  a  Un	[        R                  X5      u  pnnnnT R&                  c  Ub  [        R                  U5      T l        T R(                  c  Ub  [        R                  U5      T l        T R*                  c
  Ub  UT l        T R,                  c
  Ub  UT l        T R.                  c
  Ub  UT l        T R0                  c
  Ub  UT l        Ub  UR&                  n	[        R                  X5      u  pnnnnT R&                  c  Ub  [        R                  U5      T l        T R(                  c  Ub  [        R                  U5      T l        T R*                  c
  Ub  UT l        T R,                  c
  Ub  UT l        T R.                  c
  Ub  UT l        T R0                  c
  Ub  UT l        T $ )<Nc                j   > UT;   a,  [        TU S 5      c  [        TU TR                  U5      5        g g g r   )getattrsetattrrE   )metadata_keymodel_card_keyr   rm   s     rq   use_model_card_metadataBMetadata.apply_metadata_heuristic.<locals>.use_model_card_metadatat  s8    !Z/GHlTX4Y4aHlJNN>4RS 5b/rt   c                  > TR                  US 5      nUc  g [        TU S 5      nUc  / n[        U[        5      (       a  UR	                  U5        O&[        U[
        5      (       a  UR                  U5        [        TX5        g r   )rE   r   r   strr   listextendr   )r   r   
tags_valuecurrent_valuer   rm   s       rq   use_array_model_card_metadataHMetadata.apply_metadata_heuristic.<locals>.use_array_model_card_metadatax  su    '^^NDA
% ', E ($&Mj#..!((4
D11!((4,>rt   r   r   r   r   r"   r    r!   r$   r)   r%   r*   r&   r+   r'   r,   r(   rk   model_authormodel_versionmodel_organizationmodel_descriptionmodel_finetunemodel_basenamemodel_size_label	model_url	model_doi
model_uuidmodel_repo_urlmodel_creator
model_type
base_modelr0   base_model_sourceshttp://https://ssh://zhuggingface.coz&https?://huggingface.co/([^/]+/[^/]+)$r	   zhttps://huggingface.co/r   zbase model entry 'z' not in a known formatr3   datasetdataset_sources)r  r  r  zdataset entry 'r-   r.   r/   r1   pipeline_tagr2   language_name_or_path)r   r   r   r   )rE   r   r   r   r   r   r0   
startswithr   r   groupr   r   r   r   r   r   r3   countr   r   r!   r    r   r$   )r   rm   rn   rj   rl   r   r   metadata_base_modelsbase_model_valuer   r	  r   model_id_componentr   r   r!   r    r   r$   metadata_datasetsdataset_value
dataset_idr  dataset_id_componentdataset_name_componenthf_name_or_paths   ``                        rq   rD   !Metadata.apply_metadata_heuristicl  s    !T T? ?( $FF3#Hh7#Iy9#NNC#M=A#J
;#J
;#L,?#L%8#L%8#M6:#$5zB
 $FL9#Hn=#I?#N4HI#M3FG#J0@A#J0@A#L2DE#L+>#L+>#M<@#$57GH $FL9#Ho>#J=z)]j-HL`dnLn (*$#->>,
}^h^l^l  nB  DH  _I  AJ  $K #/!"2C88,334DE#$4d;;,334DE''/+-H( 4H!#J!(C00#..y99X=P=PQ[=\=\`h`s`st|`}`}5=z2  08;(*1Z\d(e#(9>Q$6 yA  yY  yY  Zl  y{$u$=hX`biku (A'L=E=Q=QRk=l
6(:'4'@EMEYEYZgEh
>(B'.':@G
9(= qy  qQ  qQ  RZ  qim5hPXZacm  9D5=5I5IJc5d
6 2,8=E=Q=QR_=`
> :&28?
9 5,8=V=b;RS`Raabc|b}9~
: 6#Hd33%-
 '9#h-H_%`a((//
;U !5X Z'9
+BFW[eFe$&! *z:>>)U_UcUcduw{U|;} ~ ,!-55)00?#M488)00?$$,(*H%"3J G!*c22%001RSS2<GJ/  0:=(*1Z\f(g#(;@;;q>$8u}  vV  vV  Wk  my  vz$r$:MU]_fhr (>'I:B:N:NOe:f'4'@BJBVBVWdBe(?'.':=D	(: nv  nN  nN  OY  [g  nhj2MXW^`j  6A2:2F2FG]2^,8:B:N:N}:] 7&25<	 2,8=S=_8OP]^_`v_w6x
 3#J55", s:6GG^%_`%%,,W5U #4X $Iy9#NNC#NNC)&&9)&.A)+{C)+zB
  'mmO<O*/D/DS/IQ/N +dl  eE  eE  FN  e]a)(HgWa==(-F-R$,$8$89R$SHM((0]5N,4,@,@,OH)$$,1E(0H%$$,1E(0H%##+0C'.H$&&.:3I*4H' !!H`h  aA  aA  BJ  aY]%h'S]}}$)B)N ( 4 45N O$$,1J(0(<(<](K%  (X-A$,!  (X-A$,!'G,?#* ""*z/E&0#rt   c                .   U R                   c   eU R                  b  UR                  U R                  5        U R                  b  UR	                  U R                  5        U R
                  b  UR                  U R
                  5        U R                  b  UR                  U R                  5        U R                  b  UR                  U R                  5        U R                  b  UR                  U R                  5        U R                  b  UR                  U R                  5        U R                  b  UR!                  U R                  5        U R"                  b  UR%                  U R"                  5        U R&                  b  UR)                  U R&                  5        U R*                  b  UR-                  U R*                  5        U R.                  b  UR1                  U R.                  5        UR3                  U R                   5        U R4                  b  UR7                  U R4                  5        U R8                  b  UR;                  U R8                  5        U R<                  b  UR?                  U R<                  5        U R@                  b  URC                  U R@                  5        U RD                  b  URG                  U RD                  5        U RH                  b  URK                  U RH                  5        U RL                  b  URO                  U RL                  5        U RP                  b  URS                  U RP                  5        U RT                  be  [W        U RT                  [X        5      (       a+  UR[                  SR]                  U RT                  5      5        OUR[                  U RT                  5        U R^                  b  URa                  U R^                  5        U Rb                  b  URe                  U Rb                  5        U Rf                  b  URi                  U Rf                  5        U Rj                  b  URm                  U Rj                  5        U Rn                  b  URq                  U Rn                  5        U Rr                  b  URu                  U Rr                  5        U Rv                  b  URy                  U Rv                  5        U Rz                  b  UR}                  U Rz                  5        U R~                  b  UR                  U R~                  5        U R                  b  UR                  U R                  5        U R                  Gb.  UR                  [        U R                  5      5        [        U R                  5       H  u  p#SU;   a  UR                  X#S   5        SU;   a  UR                  X#S   5        SU;   a  UR                  X#S   5        SU;   a  UR                  X#S   5        SU;   a  UR                  X#S   5        SU;   a  UR                  X#S   5        SU;   a  UR                  X#S   5        S	U;   a  UR                  X#S	   5        S
U;   d  M  UR                  X#S
   5        M     U R                  Gb.  UR                  [        U R                  5      5        [        U R                  5       H  u  p$SU;   a  UR                  X$S   5        SU;   a  UR                  X$S   5        SU;   a  UR                  X$S   5        SU;   a  UR                  X$S   5        SU;   a  UR                  X$S   5        SU;   a  UR                  X$S   5        SU;   a  UR                  X$S   5        S	U;   a  UR                  X$S	   5        S
U;   d  M  UR                  X$S
   5        M     U R                  b  UR                  U R                  5        U R                  b  UR                  U R                  5        g g )N,r   r   r   r   r"   r%   r&   r'   r(   )_r   r   add_sampling_sequencer   add_sampling_top_kr   add_sampling_top_pr   add_sampling_min_pr   add_sampling_xtc_probabilityr   add_sampling_xtc_thresholdr   add_sampling_tempr   add_sampling_penalty_last_nr   add_sampling_penalty_repeatr   add_sampling_mirostatr   add_sampling_mirostat_taur   add_sampling_mirostat_etaadd_namer   
add_authorr   add_versionr   add_organizationr    add_finetuner!   add_basenamer"   add_descriptionr#   add_quantized_byr$   add_size_labelr-   r   r   add_licenser   r.   add_license_namer/   add_license_linkr%   add_urlr&   add_doir'   add_uuidr(   add_repo_urlr)   add_source_urlr*   add_source_doir+   add_source_uuidr,   add_source_repo_urlr0   add_base_model_countr   r   add_base_model_nameadd_base_model_authoradd_base_model_versionadd_base_model_organizationadd_base_model_descriptionadd_base_model_urladd_base_model_doiadd_base_model_uuidadd_base_model_repo_urlr3   add_dataset_countadd_dataset_nameadd_dataset_authoradd_dataset_versionadd_dataset_organizationadd_dataset_descriptionadd_dataset_urladd_dataset_doiadd_dataset_uuidadd_dataset_repo_urlr1   add_tagsr2   add_languages)selfgguf_writerkeybase_model_entrydataset_entrys        rq   set_gguf_meta_modelMetadata.set_gguf_meta_model^  s   yy$$$!!---d.D.DE***4+>+>?***4+>+>?***4+>+>?((444T5R5RS&&22243N3NO)))$*<*<=''333D4P4PQ''333D4P4PQ!!---d.D.DE%%111$2L2LM%%111$2L2LMTYY';;"""4;;/<<###DLL1((():):;==$$$T]]3==$$$T]]3'''(8(89((():):;??&&&t7<<#$,,--''(>?''5((():):;((():):;88)88)99   +==$$$T]]3??&&&t7??&&&t7'''(8(89+++D,@,@A',,S1A1A-BC)243C3C)D%--33C&9QR//55cH;UV 0066sY<WX!%55;;CR`Aab $44::3Q^@_`,,2238OP,,2238OP--33C&9QR!1177j=YZ% *E( ==$))#dmm*<=&/&>"]*00F6KL},223h8OP-33Cy9QR!]288N>[\ M177==YZM)//55IJM)//55IJ]*00F6KL.44S
:ST% '?( 99   +>>%%%dnn5 &rt   r   )NNNr   )
ri   Optional[Path]rj   r_  rk   r   rl   intreturnr   r   )ri   r_  ra  dict[str, Any])rj   r_  ra  rb  )Nr   )r   r   rl   r`  ra  zMtuple[str | None, str | None, str | None, str | None, str | None, str | None])r   r   rm   Optional[dict]rn   rc  rj   r_  rl   r`  ra  r   )rY  zgguf.GGUFWriter)4__name__
__module____qualname____firstlineno__r   __annotations__r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   staticmethodrr   rF   rA   rB   rC   r   r   rD   r]  __static_attributes__r   rt   rq   r   r      s,    (,}+$(NM(&*NO*&*NO*04o4.2O2%)M?)-1]1/3_3'+}+-1?1-1?1 D- FM !G]!"&L-&"Hm""Hm"!%K%"&L-& $J$CCD-"Hm" $J$ $J$!%K%%)O])!G]!"&L-&"&L-&(,K%, $D
$%)I")%)H")U Un     - -^ 
  
     ] ] za zax o ob}6rt   r   )
__future__r   r   r|   r   loggingpathlibr   typingr   r   r   dataclassesr   	constantsr
   gguf	getLoggerr   r   r   rt   rq   <module>rs     sR    " 	     ) ) !  			:	& H6 H6 H6rt   