a ==ic @sUddlZddlZddlmZddlmZddlmZmZm Z m Z ddl m Z ddZ dd Zd d ZiZeed <iZeed <iZeed<iZeed<dS)N)create_pickler)PackageUnpickler) sys_importerOrderedImporterPackageImporterImporter)_maybe_decode_asciicsggt|tjjr|nd}|dur2t|tntfdd}t}t|}||_ | || }||r|j ndfS)Ncst|st|tjjrbt|tjjr4|j}|j}n |}tj}||dt dfSt |drt t |durdt |g|Rt t |<t t |SdS)Nstorage__reduce_deploy__ reduce_deploy)torchZ is_storage isinstancer _TypedStorage_storagedtypeZuint8appendlenhasattr_serialized_reducesgetidr )objr r importersserialized_dtypesserialized_storages^/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/torch/_deploy.py persistent_ids$    z%_save_storages..persistent_id) rr packagerrrioBytesIOrrdumpgetvalue zip_reader)importerrrZdata_bufZpicklerZ data_valuerrr_save_storagess   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 rZ_untyped_loaded_reduces _raw_packages)Zsaved_idtypenamedatar rZ reduce_idfuncargsrrr%rrpersistent_load7s      z'_load_storages..persistent_load) r _get_packagerrr!r"r0load_deploy_objects) rr%Z obj_bytesrrr0r&Z unpicklerresultrr/r_load_storages5sr5cCs|tvrt|t|<t|S)N)r*r)r%rrrr1Ys r1r*r3rr))r!r Ztorch.package._package_picklerrZ torch.package._package_unpicklerrZ torch.packagerrrrZtorch.serializationrr'r5r1r*dict__annotations__r3rr)rrrrs    -$