o 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}|durt|tntfdd}t}t|}||_ | || }||rA|j fSdfS)Ncst|s t|tjjr,t|tjjr|j}ntj}||dtdfSt |drPt t |durJdt |g| Rt t |<t t |SdS)Nstorage__reduce_deploy__ reduce_deploy)torch is_storage isinstancer TypedStoragedtypeuint8appendlenhasattr_serialized_reducesgetidr )objr importersserialized_dtypesserialized_storagesa/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/_deploy.py persistent_ids     z%_save_storages..persistent_id) rr packagerrrioBytesIOrr dumpgetvalue zip_reader)importerrr data_bufpickler data_valuerrr_save_storages s&   r+c sVfdd}durttt}nt}t|t|}||_|}t|<|S)Ncst|tsJt|d}|dd}|dkr-|d}|d}tjj||dS|dkrJ|\}}}|tvrF|tg|Rt|<t|SdS)Nrr r ) wrap_storagerr ) rtuplerr r runtyped_loaded_reduces _raw_packages)saved_idtypenamedatar r reduce_idfuncargsrrr&rrpersistent_load=s     z'_load_storages..persistent_load) r _get_packagerrr"r#r8load_deploy_objects) rr& obj_bytesrrr8r' unpicklerresultrr7r_load_storages<sr?cCs|tvr t|t|<t|S)N)r0r)r&rrrr9_s r9r0r;rr/)r"r torch.packagerrrrtorch.package._package_picklerr torch.package._package_unpicklerrtorch.serializationrr+r?r9r0dict__annotations__r;rr/rrrrs    1#