jTUddlZddlZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9ddl:Z:ddl;me?a@e7e9e4e8e4eAffeBd <d e6e9e4e8e4eAfffd ZCd e6e9e4e8e4eAfffdZDdZEde6e9e4e8e4eAfffdZFGddZGdZHdZIejJddZKde4d e8eAeAffdZLd e7eAfdZMGddZNdddeAfd ZOdS)!N)encode)Counter OrderedDict)%APPENDAPPENDSBINFLOATBINGETBININTBININT1BININT2 BINPERSIDBINPUT BINUNICODEBUILD bytes_types decode_long EMPTY_DICT EMPTY_LIST EMPTY_SET EMPTY_TUPLEGLOBALLONG1 LONG_BINGET LONG_BINPUTMARKNEWFALSENEWOBJNEWTRUENONEPROTOREDUCESETITEMSETITEMSSHORT_BINSTRINGSTOPTUPLETUPLE1TUPLE2TUPLE3UnpicklingError)unpack)maxsize)AnyCallableDictListSetTupleUnion)IMPORT_MAPPING NAME_MAPPING)sysosposixnt_marked_safe_globals_set safe_globalscTtt|adSN)r:unionsetr;s NC:\Projects\Historical\venv\Lib\site-packages\torch/_weights_only_unpickler.py_add_safe_globalsrBYs$7==c,>O>OPPreturnc*ttSr=)listr:rCrA_get_safe_globalsrH^s ( ) ))rCc"tadSr=)r?r:rGrCrA_clear_safe_globalsrJcs"uurCglobals_to_removec4tt|z adSr=)r:r?)rKs rA_remove_safe_globalsrMhs 8#>O:P:PPrCcReZdZdeeeeeefffdZdZ dZ dS) _safe_globalsr;c||_dSr=r@)selfr;s rA__init__z_safe_globals.__init__ps(rCc.t|jdSr=)rBr;)rQs rA __enter__z_safe_globals.__enter__ss$+,,,,,rCc.t|jdSr=)rMr;)rQtypevaluetbs rA__exit__z_safe_globals.__exit__vsT./////rCN) __name__ __module__ __qualname__r0r3r.r2strrRrTrYrGrCrArOrOose)T%%# :N0N*O%P))))---00000rCrOcti}tD]}t|tr~t|dkrt dt|t |dt ur%tdt |d|\}}|||<|j|j }}|||d|<|S)NzYExpected tuple of length 2 (global, str of callable full path), but got tuple of length: zHExpected second item in tuple to be str of callable full path, but got: .) r: isinstancetuplelen ValueErrorrVr] TypeErrorr[rZ)rcfnamemodules rA_get_user_allowed_globalsrksB %'' a   '1vv{{ xpstupvpvxxAaDzz$$k_cdefgdh_i_ikkGAtBtHH<DF%&B&!!4!! " " IrCc tjjtjjtjjtjjtjjtjjtjjtjj tjj tjj tjj tjj h Sr=)torch_utils_rebuild_parameter_rebuild_parameter_with_state_rebuild_qtensor_rebuild_tensor_rebuild_tensor_v2_rebuild_tensor_v3_rebuild_sparse_tensor_rebuild_meta_tensor_no_storage_rebuild_nested_tensor_rebuild_wrapper_subclass!_rebuild_device_tensor_from_numpy&_rebuild_device_tensor_from_cpu_tensorrGrCrA_tensor_rebuild_functionsr{sp ' 2 % $ ' ' + 4 + .  6  ;# rCr`c tttjjtjjtjtjtj ttttd }tjD]}||t#|<tjD]}||t#|<tjD]}|||jd|j<tjD]q}|tjjtjjfvr7tj|j||jd|j<]|||jd|j<rtjtjtjtjtjfD]}||t#|<t?D]}||d|j<tj j!|d<|S)N) zcollections.OrderedDictzcollections.Counterztorch.nn.parameter.Parameterztorch.serialization._get_layoutz torch.Sizez torch.Tensorz torch.devicez_codecs.encodezbuiltins.bytearrayz builtins.setzbuiltins.complexraz torch._utils.z#torch._tensor._rebuild_from_type_v2)"rrrmnn Parameter serialization _get_layoutSizeTensordevicer bytearrayr?complexstorage_dtype_to_storage_type_mapkeysr] _new_dtypes_tensor_classesr[rZ_storage_classes TypedStorageUntypedStorage StorageTypeper_tensor_affineper_tensor_symmetricper_channel_affineper_channel_symmetric per_channel_affine_float_qparamsr{_tensor_rebuild_from_type_v2)rgttttsqtrhs rA_get_allowed_globalsrs$/&(-(:+0+>+Jj   '#  B] 5 5 7 7 < < > >3q66 ] & & ( (3q66 #22/1bm + +bk + +,,$66 em0%-2NO O O383F3R3R 44B"-//"+// 0 046B"-//"+// 0 0  "   # . 3r77 & ( (--+, '1: ' '((16 0SB,- IrCreadlinec|ddd}|ddd}||ftvrt||f\}}n|tvr t|}||fS)Nutf-8)decoder5r4)rrjris rA_read_global_instructionrs XZZ_ # #G , ,F 8::crc? ! !' * *D ~%%#VTN3  > ! !' 4<rCc\t}d}|j}|j}itddtddt ddt ddtddtddtddtddtddtddtddtddtddt ddt"ddt$ddt&ddt(ddt*ddt,ddt.ddt0ddt2ddt4ddt6ddt8ddt:ddt<ddi } |d}|st>tA|tBsJ|dtDdkr.tG|\}}|$|d|n|d|vr||d} | r || n|dtJdkrEtMd|dd} | tNkrtQd || n|dtRdtTdhvr|dd} || nZ|dtVdkr|dd}n1|dtXdkr|StQd |d) Nrr`r_TrartA| dkr| \} } | r| j| | r+| !D]\} }tE| | | ntdt/| |dtFdkrr|j}|jd}t/|tHurtdt/||| n+|dtJdkrm|&}|jd}t/|tHurtdt/||'|n|dtPdkrD|j|j} }||jd| <nL|dtRdkrV|&}tUdtA|dD]!}||dz|jd||<"n|dtVdkr9|j|jg|_|jj|_n|dtXdkr8|&}|t?|n@|dtZdkr|jdf|jd<n|dt\dkr'|jd|jdfg|jdd<n|dt^dkr3|jd|jd|jdfg|jdd<n|dt`dkr|dn[|dtbdkr|dn-|dtddkr|dn|dtfdkr|dn|dthdkr|gn|dtjdkr|inu|dtldkr#|ton;|dtpdkr4|tsd|ddn|dttdkr0||ddn|dtvdkr4|tsd|ddn^|dtxdkr>|tsd |d!dn |dtzdkrjtsd"|dd}|t|krtd#t||d$d%}||n|dtdkrY|dd}||}|jAd&kr|B|jAd'}||n|dtdkr|j}t/|t>ur5t/|turtd(t/|t/|t>urTtA|dkrAt jEF|dd)krtd*|d||G|n |dtdtdfvrc|dtdkr |dntsd"|dd}||jJ|n|dtdtdfvrl|dtdkr |dntsd"|dd}|dkrtd+|jd|jJ|<n|dtdkr?|dd}||}|t|n|dtdkr?|dd|_Q|jQdkrtjSd,|jQd-nJ|dtdkr|j}|Std.|dq)/zRead a pickled object representation from the open file. Return the reconstituted object hierarchy specified in the file. Tr`rraz"Trying to load unsupported GLOBAL z whose module z is blocked.)z1torch.nested._internal.nested_tensor.NestedTensorz1torch.nested._internal.nested_tensor._rebuild_njtz"torch._dynamo.decorators._DimRangez\``torch.nested`` and ``torch._dynamo`` must be imported to load nested jagged tensors (NJTs))z(torch.distributed.device_mesh.DeviceMeshz2torch.distributed.tensor._dtensor_spec.DTensorSpecz1torch.distributed.tensor._dtensor_spec.TensorMetaz torch.distributed.tensor.DTensorz0torch.distributed.tensor.placement_types.Partialz2torch.distributed.tensor.placement_types.Replicatez.torch.distributed.tensor.placement_types.Shardz>``torch.distributed.tensor`` must be imported to load DTensorszUnsupported global: GLOBAL zY was not an allowed global by default. Please use `torch.serialization.add_safe_globals([z.])` or the `torch.serialization.safe_globals([zN])` context manager to allowlist this global if you trust this class/function.zbCan only create new object for nn.Parameter or classes allowlisted via `add_safe_globals` but got rz0Trying to call reduce for unrecognized function __setstate__Nr_zcCan only build Tensor, Parameter, OrderedDict or types allowlisted via `add_safe_globals`, but got z"Can only append to lists, but got zCan only extend lists, but got FrGzdrrrr surrogatepassrstrictz1persistent_load id must be tuple or int, but got rz4Only persistent_load of storage is allowed, but got znegative argumentzDetected pickle protocol z in the checkpoint, which was not the default pickle protocol used by `torch.load` (2). The weights_only Unpickler might not support all instructions implemented by this protocol, please file an issue for adding support if you encounter this.r)U metastackstackappendrrrrbrrr_blocklisted_modulesr*rrkrpoprmr}r~values__new__r!rrVrset_rr__dict__updatehasattrrcrditemssetattrrrFrpop_markextendr"r#rangerr&r'r(r)rrrrrrrr?r r+r r rrr,r]r$rrr intr_maybe_decode_asciipersistent_loadr rrrrrerrr rwarningswarnr%)rQrrrrjri full_pathargsclsfuncstateinst slotstatekvitemlist_objrirstrvalstrdatapididxndatargs rAloadzUnpickler.load5s  " j' y=\ G$q''C c;// / //1v""7 FF %.... 111)jYjjV\jjj 4 6 666KK 4 6 6y ABBBB";"="===KK 9 ; ;I FGGGG*v  *X*B<<< Q6!9$$z~~''jnn&&%(,,,KK 2D 9::::466==????244;;====KK  C 7$ 7 7 78888)@:=@@Q6!9$$z~~''z"~ 4 6 6 = = ? ???$=$?$?$F$F$H$HHH)Q4QQ"&t 2Q58## ((z"~::--DIu%%%$ZZ58#555%%e,,,,$ZZ;..M((////JJ";"="="D"D"F"FFFDzz%9%;%;%B%B%D%DDDt^44 4))%0000%) %eU335E a/4,E9 8 M00777$4(1(9(9441 'a 3 3 3 3)H;?::HH Q6!9$$z~~'':b>>>--)MT(^^MM%%%%Q71:%% :b>>>--)J$x..JJ&&&&Q71:%%*..**DJNN,<,)) BQ:a=(( BQ:a=(( BQ9Q<'' CEE""""Q6!9$$ F4a11!45555Q71:%% DIIaLLO,,,,Q71:%% F4a11!45555Q8A;&& F4166q9::::Q:a=((dd1gg..q1G##)*>???TT&\\7ODD F####Q?1---a$v,,=G++%nnT]HEEG G$$$$Q9Q<''jnn&&99E))$s))32F2F)WDQTIIWWII&&C1 +??AGG9TT)WsSTvWW D00556666QF1I{1~666"%a&F1I"5"5ttAwww6$Q;P;PRST DIcN++++QF1I{1~666 #A&) 3 3TT!WWWdd1gg9N9NPQRq55$%8999#z"~ ! Q58##DGGAJtAww K--....Q58##!T!WWQZ :??MYDJYYY Q47""Z^^%% %&ESV&E&EFFFy\ GrCcr|j}|j|_|jj|_|Sr=)rrrr)rQrs rArzUnpickler.pop_marks/ ^'')) j'  rCc td)Nz%unsupported persistent id encountered)r*)rQrs rArzUnpickler.persistent_load$sEFFFrCN)rZr[r\r]rRrrrrGrCrArr-su07#fGfGfGR GGGGGrCrASCIIrrcHt||S)Nr)rr)rrs rArr(s! TH - - - 2 2 4 44rC)P functools _functoolsr_codecsr collectionsrrpicklerrrr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*structr+r6r,typingr-r.r/r0r1r2r3rm torch._utilsr4r5rr?r:r]__annotations__rBrHrJrMrOrkr{ lru_cacherrrrrrGrCrArst2,,,,,,,,&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&N?????????????????? 55555555 HKsuu#eHeHcM.B$BCDLLLQDxx}9M/M)N$OQQQQ *4hhm0D&D EF**** %%% QE(E(C-,@"@ABQQQQ00000000$(.a   00! 0f x E#s(O    >CC>C>C>C>CBxGxGxGxGxGxGxGxGv#*555C555555rC