a h @sUddlZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddZ dd Zd d ZiZeed <iZeed <iZeed<iZeed<dS)N)ImporterOrderedImporterPackageImporter sys_importer)create_pickler)PackageUnpickler)_maybe_decode_asciicsggt|tjjr|nd}|dur2t|tntfdd}t}t|}||_ | || }||r|j ndfS)Ncst|st|tjjrXt|tjjr.|j}ntj}||dtdfSt |drt t |durdt |g| Rt t |<t t |SdS)Nstorage__reduce_deploy__ reduce_deploy)torchZ is_storage isinstancer TypedStoragedtypeZuint8appendlenhasattr_serialized_reducesgetidr )objr importersserialized_dtypesserialized_storages;/var/www/auris/lib/python3.9/site-packages/torch/_deploy.py persistent_ids     z%_save_storages..persistent_id) rr packagerrrioBytesIOrrdumpgetvalue zip_reader)importerrrZdata_bufZpicklerZ data_valuerrr_save_storages s"    r&c sVfdd}dur(ttt}nt}t|t|}||_|}t|<|S)Ncst|tsJt|d}|dd}|dkrZ|d}|d}tjj||dS|dkr|\}}}|tvr|tg|Rt|<t|SdS)Nrr r )Z wrap_storagerr ) rtuplerr r rZuntyped_loaded_reduces _raw_packages)Zsaved_idtypenamedatar rZ reduce_idfuncargsrrr$rrpersistent_load=s     z'_load_storages..persistent_load) r _get_packagerrr r!r/load_deploy_objects) rr$Z obj_bytesrrr/r%Z unpicklerresultrr.r_load_storages<sr4cCs|tvrt|t|<t|S)N)r)r)r$rrrr0_s r0r)r2rr()r r Z torch.packagerrrrZtorch.package._package_picklerrZ torch.package._package_unpicklerrZtorch.serializationrr&r4r0r)dict__annotations__r2rr(rrrrs    1#