qyi1' SSKrSSKrSSKrSSKrSSKJrJr SSKJr SSK J r \R"S5r S\ S\ 4SjrS$S \\S\ 4S jjrS \\ \ 4S\\ \ 44S jr\R"S 5rS\ S\ 4SjrSrSrSSSSSSSSS.rSrS%Sjr\"SS55rS&S\ S \S!\S"\R:S-4S#jjrg)'N) OrderedDict defaultdict) dataclass)Anyz(?<=\.)(\d+)(?=\.|$)keyreturnc.[RSU5$)zBReplace every dot-delimited integer with '*' to get the structure.*) _DIGIT_RXsub)rs ^/home/mande/repo/quber/.venv/lib/python3.13/site-packages/transformers/utils/loading_report.py _pattern_ofrs ==c ""valuesc[U5S:Xa[US5$[U5n[U5U:a USSUS3$SR[ [U55$)zEFormat a list of ints as single number, {a, ..., b}, or first...last.rz...z, )lenstrsortedjoinmap)rcutoffs r _fmt_indicesrs` 6{a6!9~ F^F 6{V)Cr |,, 99Sf% &&rmappingc Sn[U[5(dUVs0sHo"U_M nnSn[[5nUR 5HupE[ R U5n[U5n[U5HPup[X75U::aX7R[55 X7UR[U 55 MR X7RU5 M 0n UR 5Hup{U SSU Sp\URS5n U Sn[S[U 55HknUS- [U 5:aJXS- (a>[![#XS- 55n[XS- 5S:a USU-S -- nO X- nOUS- nXU- nMm XZU'M [%U 5nU(aUR'5$U$s snf) 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}. FTNrr rr{}) isinstancedictrlistitemsr findallr enumeraterappendsetaddintsplitrangerrrkeys)r not_mappingkbucketrvaldigspattid out_itemsrsetspartsfinalinsertouts r update_key_namer:)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 G z\x1b\[[0-9;]*msc@[RS[U55$)N)_ansi_rer r)r;s r _strip_ansir?Ys <<CF ##rc j[U5n[SU[[U55- 5nUSU--$)Nr )rmaxrr?)textwidthtpads r _padrG]s3 D A a[^,, -C sSy=rc U(a[[U/U-65OU/nUVs/sHn[SU55PM nnSRS[X555nSRSU55nUVs/sH$nSRS[Xt555PM& nnSRXV/U-5$s snfs snf)Nc3J# UHn[[U55v M g7fN)rr?).0xs r _make_table..fs3s!#k!n%%ss!#z | c3<# UHup[X5v M g7frJrG)rKhws r rMrNgsI4HDAT!ZZ4Hz-+-c3,# UH nSU-v M g7f)-N)rKrRs r rMrNhs26a#'6sc3<# UHup[X5v M g7frJrP)rKcrRs r rMrNis=ndatAzznrS )r!ziprBr) rowsheaderscolscolwidths header_linesep_linerbodys r _make_tablerdcs-14wi$&( )yD?C Dtc3s33tF D**IC4HIIKzz2622HGK Lt!EJJ=c!n= =tD L 99k,t3 44 E Ms C?+Czzzz zzzz)resetredyelloworangepurplebolditalicdimcx[RR5(a[UU[S3$U$)zdReturn color/style-formatted input `s` if `sys.stdout` is interactive, e.g. connected to a terminal.re)sysstdoutisattyPALETTE)r;colors r _stylersys8 zz%.!!WW%5$677rcf[R"5R$![a Us$f=frJ)shutilget_terminal_sizecolumns Exception)defaults r _get_terminal_widthrzs0'')111 s ! 00c\rSrSr%Sr\\\S'\\\S'\\\\\ \\ 4\S'\ \\S'\ \\4\S'Sr S r S \S -4S jrS rg )LoadStateDictInfoa2 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. 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. missing_keysunexpected_keysmismatched_keys error_msgsconversion_errorscdURURVs1sHoSiM sn-$s snf)zMReturn all effective missing keys, including `missing` and `mismatched` keys.r)r~r)selfr-s r missing_and_mismatched(LoadStateDictInfo.missing_and_mismatcheds/  $2F2F#G2FQaD2F#GGG#Gs-c`URURURURS.$)Nr~rrrr)rs r to_dictLoadStateDictInfo.to_dicts0!--#33#33//   rrNc [5n/nSnUR(a\US[SS5S3[SS5-- n[UR5H#n[SS5nUR XES S /5 M% UR (a[US[S S 5S3[S S5-- n[UR 5H"n[S S 5nUR XES /5 M$ UR (aUS[SS5S3[SS5-- nUR VVVs0sH upgoXx4_M n nnn[U 5R5HBun up[SS5nU US[U 5S[U 53/n UR U 5 MD UR(aqUS[SS5S3[SS5-- n[UR5R5H*upN[SS5nSUS3nUR XEU/5 M, [U5S:XagSS/nUS:aUS/- nOUS S /- n[UUS9nUU-nU$s snnnf) z/Generate the minimal table of a loading report.z Notes:z - z UNEXPECTED:rh zbcan be ignored when loading from different task/architecture; not ok if you expect identical arch.rk UNEXPECTEDr=zMISSING:rfzsthose params were newly initialized because missing from the checkpoint. Consider training on your downstream task.MISSINGz MISMATCH:rgzRckpt 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:riz$originate from the conversion scheme CONVERSIONz rNKeyStatusDetails)r\) rzrrsr:r%r~rr"rrrrd)rterm_wr[tipsr-statusabrXiteratorr shape_ckpt shape_modeldatav_detailsr\tablereports r create_loading_report'LoadStateDictInfo.create_loading_reports$&    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 t9>(# C<  { "G Bx GD'2 ?Gs)IrV)__name__ __module__ __qualname____firstlineno____doc__r&r__annotations__tupler(r!r rrr__static_attributes__rVrr r|r|so"c(XsE#Jc :;<<S CH~%H ;sTz;rr|pretrained_model_name_or_pathignore_mismatched_sizes loading_infologgerc"Uc[R"[5nUR(aKSR UR5nSU;aUS- n[ SUR RSU35eUR5nUcg[SUR RS[S S US 3nURXv-5 UR(a [ S 5eU(dUR(a [ S 5egg)z Log a readable report about state_dict loading issues. This version is terminal-size aware: for very small terminals it falls back to a compact Key | Status view so output doesn't wrap badly. Nz z size mismatchzb You may consider adding `ignore_mismatched_sizes=True` to `from_pretrained(...)` if appropriate.z#Error(s) in loading state_dict for z: rjz LOAD REPORTrez from: rYzWe encountered some issues during automatic conversion of the weights. For details look at the `CONVERSION` entries of the above report!zjYou set `ignore_mismatched_sizes` to `False`, thus raising an error. For details look at the above report!) logging getLoggerrrr RuntimeError __class__rrqwarningrr)modelrrrr error_msgrpreludes r log_state_dict_reportrs ~""8,KK 7 78 i ' v I@AYAY@ZZ_`i_jkll / / 1F ~!%//":":!;<PWHXGYY`a~`@BCG NN7#$%%    #|'C'C x  (D "r) )PrJ)rrerurn collectionsrr dataclassesrtypingrcompiler rrr!r(rr r:r>r?rGrdrqrsrzr|boolLoggerrrVrr rs* 0! JJ. / #S#S# 'c'#'*T#s(^*S#X*Z ::' ($3$3$ 5     `` `P%) , #&, ", $ , NNT ! , r