#h 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)Nc|tj|st|tjjrzt|tjjr|j}|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_valuerrr s @@@r!_save_storagesr. s%h 0MNNXxxTXH#Hl;;  6z||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_idfuncargsrr r)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_bytesr rr<r* unpicklerresults ` `` r!_load_storagesrC=s."< #;#;\JJ 2:i+@+@AAI /I#,>>#3#33F_R Mr#c\|tvrt|t|<t|S)N)r5r)r)s r!r=r=`s*&&$3J$?$? j!  $$r#r5r?rr4)r%r torch.packagerrrrtorch.package._package_picklerr torch.package._package_unpicklerrtorch.serializationr r.rCr=r5dict__annotations__r?rr4r#r!rLs RRRRRRRRRRRR999999======333333000f   F%%%  tTr#