U Kc(@sddlZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.m/Z/m0Z0ddl1Z1ej2dddd Z3Gd d d Z4d d e5d ddZ6dS)N) OrderedDict)$APPENDAPPENDSBINGETBININTBININT1BININT2 BINPERSIDBINPUT BINUNICODEBUILD bytes_types decode_long EMPTY_DICT EMPTY_LIST EMPTY_SET EMPTY_TUPLEGLOBALLONG1 LONG_BINGET LONG_BINPUTMARKNEWFALSENEWOBJNEWTRUENONEPROTOREDUCESETITEMSETITEMSSHORT_BINSTRINGSTOPTUPLETUPLE1TUPLE2TUPLE3UnpicklingError)unpack)maxsize)AnyDictListc Csttjjtjjtjtjd}tjtj tj tj tj tj tjtjtjtjf D]}||t|<qHtjD]}|||jd|j<q`tjD]}|||jd|j<qtjjtjjtjjtjjtjjtjjfD]}||d|j<q|S)N)zcollections.OrderedDictztorch.nn.parameter.Parameterztorch.serialization._get_layoutz torch.Sizez torch.Tensor.z torch._utils.)rtorchnn Parameter serializationZ _get_layoutSizeTensorZ complex32Z complex64Z complex128Zfloat16Zfloat32Zfloat64Zint8Zint16Zint32Zint64strZ_tensor_classes __module____name__Z_storage_classes_utilsZ_rebuild_parameterZ_rebuild_tensorZ_rebuild_tensor_v2Z_rebuild_sparse_tensorZ_rebuild_meta_tensor_no_storageZ_rebuild_sparse_csr_tensor)rcttttsfr=A/tmp/pip-unpacked-wheel-gikjz4vx/torch/_weights_only_unpickler.py_get_allowed_globalsCs>    r?c@s8eZdZddedddZddZddZd d Zd S) UnpicklerbytesencodingcCs ||_|j|_|j|_i|_dSN)rCreadlinereadmemo)selffilerCr=r=r>__init__nszUnpickler.__init__cCsg|_g|_|jj|_|j}|j}|d}|s2tt|ts@t|dt dkr|dd d}|dd d}|d|}|t kr|t |nt d|q"|dt dkr|j}|j}|tjjk rt d||tjj|q"|dtdkrZ|j}|jd} | t krJt d | | ||jd<q"|dtdkr|j} |jd} t| tjkr| j| nLt| tjjkr| | n.t| tkr| j| nt d t| q"|dtdkr<|j} |jd} t| tk r0t d t| | | q"|dtdkr|}|jd} t| tk rt d t| | |q"|dt!dkr|j|j}}||jd|<q"|dt"dkr|}t#dt$|d D] }||d|jd||<qq"|dt%dkrD|j|jg|_|jj|_q"|dt&dkrn|}|t'|q"|dt(dkr|jdf|jd<q"|dt)dkr|jd|jdfg|jdd<q"|dt*dkr|jd|jd|jdfg|jdd<q"|dt+dkr"|dq"|dt,dkr@|dq"|dt-dkr^|dq"|dt.dkr||dq"|dt/dkr|gq"|dt0dkr|iq"|dt1dkr|t2q"|dt3dkr|t4d|ddq"|dt5dkr,||ddq"|dt6dkrX|t4d|d dq"|dt7dkrt4d|dd}|t8krt dt9||dd}||q"|dt:dkr|dd}||}|j;dkr| |j;d}||q"|dt?|ddkrt d|d||@|q"|dtAdtBdfkr|dtAdkr|dn t4d|dd}||jC|q"|dtDdtEdfkrX|dtDdkr"|dn t4d|dd}|dkrFtFd|jd|jC|<q"|dtGdkr|dd}||}|tH|q"|dtIdkr|ddq"|dtJdkr|j}|St d|dq"dS) zRead a pickled object representation from the open file. Return the reconstituted object hierarchy specified in the file. r,rNzutf-8r-zUnsupported class z(Trying to instantiate unsupported class z0Trying to call reduce for unrecognized function z:Can only build Tensor, parameter or dict objects, but got z"Can only append to lists, but got zCan only extend lists, but got FTr=zloadts                      "*                ,,   zUnpickler.loadcCs |j}|j|_|jj|_|SrD)rSrRrZrT)rHrvr=r=r>rbs  zUnpickler.pop_markcCs tddS)Nz%unsupported persistent id encountered)r&)rHr{r=r=r>riszUnpickler.persistent_loadN)r6r5 __qualname__r4rJrrbrir=r=r=r>r@ms "r@ASCIIrBcCst||dS)NrB)r@r)rIrCr=r=r>rsr)7 functools _functools collectionsrpicklerrrrrrr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&structr'sysr(typingr)r*r+r. lru_cacher?r@r4rr=r=r=r>s &    )3