+ j.T^RIt^RIt^RIt^RIt^RIHtHt^RIHt^RI H t ^RI H t ]P!R4tRRltR,R R lltR R lt]P!R 4tRRltRtRtRRRRRRRRRRRRRRR R!R"R#/ tR$tR-R%ltR&R'lt]!R(R)44tR.R*R+lltR#)/N) OrderedDict defaultdict) dataclass)Any) PipelineStagez(?<=\.)(\d+)(?=\.|$)c0V^8dQhR\R\/#)keyreturnstr)formats"b/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/utils/loading_report.py __annotate__rs##S#S#c.\PRV4#)zBReplace every dot-delimited integer with '*' to get the structure.*) _DIGIT_RXsub)r s&r _pattern_ofrs ==c ""rcFV^8dQhR\\,R\/#)r valuesr )listintr )rs"rrr!s''c'#'rc\V4^8Xd\V^,4#\V4p\V4V8dV^, RVR, 2#RP\ \V44#)zEFormat a list of ints as single number, {a, ..., b}, or first...last.z...z, )lenr sortedjoinmap)rcutoffs&&r _fmt_indicesr"!s` 6{a6!9~ F^F 6{V)Cr |,, 99Sf% &&rctV^8dQhR\\\3,R\\\3,/#)r mappingr )dictr r)rs"rrr+s***T#s(^*S#X*rc TRp\V\4'gVUu/uFq"VbK ppRp\\4pVP 4FwrE\ P V4p\V4p\V4Fawr\W7,4V8:d W7,P\44W7,V,P\V 44Kc W7,PV4K /p VP 4Fwr{V RRV R,r\VPR4p V ^,p\^\V 44FpV^, \V 48duW^, ,'d`\!\#W^, ,44p\W^, ,4^8dVRV,R,, pMW, pM VR, pWV,, pK WZV&K \%V 4pV'dVP'4#V#uupi)z Merge keys like 'layers.0.x', 'layers.1.x' into 'layers.{0, 1}.x' BUT only merge together keys that have the exact same value. Returns a new dict {merged_key: value}. FTNr{}r) isinstancer%rritemsrfindallr enumeraterappendsetaddrsplitranger"rrkeys)r$ not_mappingkbucketr valdigspattid out_itemsrsetspartsfinalinsertouts& rupdate_key_namerA+sK gt $ $!()Aa4) .9$.?FMMO  %3dODA6< A% ##CE* LO  A '$  C $I  3BKc 3aq#e*%A1us4y Ta%[[%fTa%[&9:tE{#a'S6\C//EOE  1X E&%' i Cxxz JC*s H%z\x1b\[[0-9;]*mc0V^8dQhR\R\/#)r sr r )rs"rrr[s$$3$3$rc@\PR\V44#))_ansi_rerr )rCs&r _strip_ansirG[s <<CF ##rc \V4p\^V\\V44, 4pVRV,,#)r )r maxrrG)textwidthtpads&& r_padrO_s3 D A a[^,, -C sSy=rc V'd\\V.V,!4MV.pVUu.uFp\RV44NK ppRPR\W444pRPRV44pVUu.uF$pRPR\Wt444NK& ppRPWV.V,4#uupiuupi)c3J"TFp\\V44xK R#5iN)rrG).0xs& r _make_table..hs3s!#k!n%%ss!#z | c3<"TFwr\W4xK R#5irRrO)rShws& rrUrVisI4HDAT!ZZ4Hz-+-c34"TFpRV,xK R#5i)-N)rSrZs& rrUrVjs26a#''6sc3<"TFwr\W4xK R#5irRrX)rScrZs& rrUrVks=ndatAzznr[ )rziprJr) rowsheaderscolscolwidths header_linesep_linerbodys && r _make_tablerles-14wi$&( )yD?C Dtc3s33tF D**IC4HIIKzz2622HGK Lt!EJJ=c!n= =tD L 99k,t3 44 E Ms C *Cresetzredzyellowzorangez purplezgreenzboldzitaliczdimzc\PeC\PP4'd\V, V \R, 2#V#)zdReturn color/style-formatted input `s` if `sys.stdout` is interactive, e.g. connected to a terminal.rm)sysstdoutisattyPALETTE)rCcolors&&r_styler||sA zz#**"3"3"5"5%.!!WW%5$677rch\P!4P# \dTu#i;irR)shutilget_terminal_sizecolumns Exception)defaults&r_get_terminal_widthrs0'')111 s ! 11c V^8dQhR\R\\,R\\\\ 3,\\\ 3,3,/#)r stageskipped_pp_keysr )rr.r tupler%r)rs"rrrsB    25c(  4S>4S> )* rcVPp\\W4P4p\P VP 4VP4pVUu/uFpVPWdV4;pfKWgbK ppWX3#uupi)z@Map checkpoint keys to their owning PP rank for the load report.) base_model_prefixrgetattrlayersr%fromkeys state_dictpp_rankfind_rank_for_key) rmodelrprefix num_layersownedr ownerskippeds &&& r_pp_report_key_ownersrs $ $FWU+223J MM%**,emm  s B:BcRa]tRt^toRtRtRtR V3RlRlltV3RltRt Vt R#) LoadStateDictInfoa Mutable container for state-dict loading results and diagnostics. Each entry in this structure is mutable, and will usually be mutated in-place during the loading pipeline. Attributes: missing_keys (`set[str]`): Keys that are missing from the loaded checkpoints but expected in the model's architecture. unexpected_keys (`set[str]`): Keys that are found in the checkpoints, but not expected in the model's architecture. skipped_pp_keys (`set[str]`): Checkpoint keys owned by another pipeline-parallel rank and intentionally not loaded here. mismatched_keys (`set[tuple[str, tuple[int], tuple[int]]]`): Keys that are found in the checkpoints and are expected in the model's architecture, but with a different shape. error_msgs ( `list[str]`): Some potential error messages. conversion_errors (`dict[str, str]`): Errors happening during the on-the-fly weight conversion process. c rVPVPUu0uF q^,kK up,#uupi)zMReturn all effective missing keys, including `missing` and `mismatched` keys.) missing_keysmismatched_keys)selfr4s& rmissing_and_mismatched(LoadStateDictInfo.missing_and_mismatcheds1  $2F2F#G2FQaDD2F#GGG#Gs4cfRVPRVPRVPRVP/#runexpected_keysr error_msgsr)rs&rto_dictLoadStateDictInfo.to_dicts8 D-- t33 t33 $//   rNc.<V^8dQhRS[R,/#)r r Nr )r __classdict__s"rrLoadStateDictInfo.__annotate__sJJ3:Jrc  \4p.pRpVP'deVR\RR4 R2\RR4,, p\VP4F#p\RR4pVP WVR R .4K% VP 'ddVR\R R 4 R2\R R4,, p\VP 4F"p\R R 4pVP WVR .4K$ VP 'dVR\RR4 R2\RR4,, pVP UUU u/uF wrxqW3bK p ppp \V 4P4FBwp wr\RR4pV VR\V 4 R\V 4 2.pVP V4KD VP'dzVR\RR4 R2\RR4,, p\VP4P4F*wr_\RR4pRV R2pVP WVV.4K, RpVe\VRR4pVe\VWP4wppRRRV3RRR V33FrwppppV'gKVR\V R!2V4 RV 2, p\V4P4F'wp pVP V \VV4R"V 2.4K) Kt \V4^8XdR#R#R$.pV^8d VR%., pM VR R ., p\VVR&7pVV,pV#uup ppi)'z/Generate the minimal table of a loading report.z Notes:z - z UNEXPECTED:rp zbcan be ignored when loading from different task/architecture; not ok if you expect identical arch.rt UNEXPECTEDrEzMISSING:rnzsthose params were newly initialized because missing from the checkpoint. Consider training on your downstream task.MISSINGz MISMATCH:rozRckpt weights were loaded, but they did not match the original empty weight shapes.MISMATCHz$Reinit due to size mismatch - ckpt: z vs model:z CONVERSION:rqz$originate from the conversion scheme CONVERSIONz N _pp_stageOWNEDrrz1checkpoint weights loaded on this pipeline stage.SKIPPEDruz3checkpoint weights owned by another pipeline stage.:zPP rank KeyStatusDetails)rd)rrr|rAr-rrr*r conversion_errorsrrrrrl)rrterm_wrctipsr4statusabr`iteratorr shape_ckpt shape_modeldatav_detailsrrrr{note key_ownersrrdtablereports&& rcreate_loading_report'LoadStateDictInfo.create_loading_reportsz$&     d6-:;2>tB D%T%9%9: h7 QB/0;     d6*e45R86F< D%T%6%67 51 QO,8     d6+x896B D((>(>?EEG h7!!D> Q12H   E;5E  25%AUAUVNE7'#VX]^E#XZab4/tZ:d6VHA,#>"?r$HHD&5j&A&G&G&I U S&*?8E7AS$TU'J 4 t9>(# C<  { "G Bx GD'2 [GsL>c<V^8dQh/S[S[,;R&S[S[,;R&S[S[S[S[S[,S[S[,3,,;R&S[S[,;R&S[S[S[3,;R&S[S[,;R&#)r rrrrrr)r.r rrrr%)rrs"rrrs*c(+,X-.sE#Jc :;<rs 0!9 JJ. / # '*Z ::' ($ 5 V 7 g m g W F f 6    rr rj- - r