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|tjjr1t|tjjr|j}|j}n|}tj}||dt dfSt |drUt t |durOdt |g|Rt t |<t t |SdS)Nstorage__reduce_deploy__ reduce_deploy)torch is_storage isinstancer TypedStorage_untyped_storagedtypeuint8appendlenhasattr_serialized_reducesgetidr )objr r importersserialized_dtypesserialized_storagesC/var/www/html/ai/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'rr persistent_load>s     z'_load_storages..persistent_load) r _get_packagerrr#r$r9load_deploy_objects) rr' obj_bytesrrr9r( unpicklerresultrr8r _load_storages=sr@cCs|tvr t|t|<t|S)N)r1r)r'rrr r:`s r:r1r<rr0)r#r torch.packagerrrrtorch.package._package_picklerr torch.package._package_unpicklerrtorch.serializationrr,r@r:r1dict__annotations__r<rr0rrrr s    3#