קg UddlZddlZddlmZmZmZmZddlmZddl m Z ddl m Z dZ dZdZiZeed <iZeed <iZeed <iZeed <dS) N)ImporterOrderedImporterPackageImporter sys_importer)create_pickler)PackageUnpickler)_maybe_decode_asciicfggt|tjjr|nd}|t |t nt fd}t j}t|}||_ | || }||r|j ndfS)Ncjtj|st|tjjrqt|tjjr|j}n tj}||dtdz fSt|dr}t t|.persistent_ids  C =JsEM4N$O$O =#u}9:: $    & &s + + +  $ $U + + +s#677!;< < 3+ , , 0"&&r#ww//7#sGG0**95500#BsGG, 'r#ww/ /t) rrpackagerrrioBytesIOrr!dumpgetvalue zip_reader) importerrr!data_bufpickler data_valuerrrs @@@r _save_storagesr- s%h 0MNNXxxTXH#Hl;;  2z||HXy11G)G LL""$$J'1T  r"cfd}#ttt}nt}t|t j|}||_|x}t|<|S)Nct|tsJt|d}|dd}|dkrO |d}|d}tj||S|dkr8|\}}}|tvr|t g|Rt|<t|SdS)Nrr r ) wrap_storagerr) rtupler rr runtyped_loaded_reduces _raw_packages) saved_idtypenamedatar r reduce_idfuncargsrrr(s r persistent_loadz'_load_storages..persistent_load=s(E*****&x{33| y *$q'2G%d1g.E=--$__..e.   & &$( !ItT//-1T- 2K-Sd-S-S-S *"9- -tr") r _get_packagerrr$r%r;load_deploy_objects) rr( obj_bytesrrr;r) unpicklerresults ` `` r _load_storagesrB<s."< #;#;\JJ 2:i+@+@AAI /I#,>>#3#33F_R Mr"c\|tvrt|t|<t|S)N)r4r)r(s r r<r<_s*&&$3J$?$? j!  $$r"r4r>rr3)r$r torch.packagerrrrtorch.package._package_picklerr torch.package._package_unpicklerrtorch.serializationr r-rBr<r4dict__annotations__r>rr3r"r rKs RRRRRRRRRRRR999999======333333...b   F%%%  tTr"