tj.UddlZddlmZddlmZddlmZddlZddlm Z ddl m Z m Z m Z mZgZeeed<eGddZe jegGd d Zee fd ed ed ededdf dZ dddd eded edededef dZdS)N) dataclass)BufferedIOBase)Any)_load_saveDEFAULT_PROTOCOL MAP_LOCATION__all__c.eZdZUeed<eed<eed<dS)_Entrykey is_storagelengthN)__name__ __module__ __qualname__str__annotations__boolint^/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/distributed/_serialization.pyr r s. HHH KKKKKrr ceZdZddZdedededdfdZdeddfd Z deddfd Z dede fd Z dedefd Z ded ededejfdZdefdZdS)_PseudoZipFilereturnNci|_dSNrecordsselfs r__init__z_PseudoZipFile.__init__s 68 rr datarc||f|j|<dSrr)r"r r$rs r write_recordz_PseudoZipFile.write_records!6N Srfc g}|jD]E\}\}}|t|t |t j|Ftj||t|j D]\}}t |tr| |0t |tr)| |dnt |t jr||dddt!dt#|dS)N)r rr)protocolutf-8Fzunknown type: )r itemsappendr isinstancetorchUntypedStoragepickledumprvaluesbyteswriterencode _write_file TypeErrortype)r"r'entriesr r$r_s rwrite_toz_PseudoZipFile.write_tosZ#'<#5#5#7#7   C$ NN)$0DEE!      GQ)9::::|**,, ? ?GD!$&& ? D#&& ? G,,----D%"677 ?  E5!4444 =d = =>>> ? ?rcxtj|}|D]}||j}|jri|jdkrt jd}n2t j|t j }||jf|j |j <||jf|j |j <dSNr)dtype) _weights_only_unpicklerloadreadrrr/r0 frombufferuint8untyped_storager r )r"r'r:entryr$storages r read_fromz_PseudoZipFile.read_from6s).q11 ? ?E66%,''D ?<1$$#2155GG#.#k&o'' L+ UY'' ,0*> UY''! ? ?rc||jvSrrr"r s r has_recordz_PseudoZipFile.has_recordKsdl""rc(|j|dS)NrrrJs r get_recordz_PseudoZipFile.get_recordNs|C ##r_length_typecdtj|j|dtjSr>)r/tensorr rD)r"r rNrOs rget_storage_from_recordz&_PseudoZipFile.get_storage_from_recordQs'|DL-a0 DDDDrcdS)Ntorchftrr!s rserialization_idz_PseudoZipFile.serialization_idVsyr)rN)rrrr#robjectrr&rr<rHrrKrMr/TensorrRrUrrrrrs99999++6+3+4++++?.?T????.?>?d????*#c#d####$c$f$$$$EE!$E-0E EEEE #rrobjr' pickle_modulepickle_protocolrctt}t||||d||dS)a5 Save the object to a file-like object in a streaming fashion compatible with network sockets. This behaves similarly to :func:`torch.save` with a few notable differences: * A non-seekable file like object can be used when loading. * No forwards/backwards compatibility is provided for the serialization format. This is only intended to be used with a single version of PyTorch with transient storage (i.e. sockets or temp files). * mmap is not supported See :func:`torch.save` for more details on specific arguments. F)zip_filerYrZ_disable_byteorder_recordN)rrr<)rXr'rYrZr\s r_streaming_saver^ZsQ*H  #'"'  arT) weights_only map_locationr_pickle_load_argsc |r|tdt}n |t}d|vrd|d<t}||t d|||d|S)a Load the object from a file-like object in a streaming fashion compatible with network sockets. See :func:`_streaming_save` for more details about the streaming behavior. See :func:`torch.load` for more details on specific arguments. NzDCan not safely load weights when explicit pickle_module is specifiedencodingr*)r\r`rYr) RuntimeErrorr@r1rrHr)r'r`rYr_rar\s r_streaming_loadrezs #  $V 0  "M)))'.$H q  !#      r)NN)r1 dataclassesriortypingrr/torch._weights_only_unpicklerr@torch.serializationrrrr r listrrr _add_safe_globalsrrVrr^rrerrrrms !!!!!! ??????LLLLLLLLLLLLc   *)6(333????????J +     D"&$  $$$$$$  $  $ $$$$$$r