U ʹj)@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(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/m0Z0m1Z1ddl2Z2ej3dddd Z4Gd d d Z5d d e6d ddZ7dS)N) OrderedDict)%APPENDAPPENDSBINFLOATBINGETBININTBININT1BININT2 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<qtjjtjjtjjtjjtjjfD]}||d|j<qtjj|d<|S)N)zcollections.OrderedDictztorch.nn.parameter.Parameterztorch.serialization._get_layoutz torch.Sizez torch.Tensor.z torch._utils.z#torch._tensor._rebuild_from_type_v2)rtorchnn Parameter serializationZ _get_layoutSizeTensorZ complex32Z complex64Z complex128Zfloat16Zfloat32Zfloat64Zint8Zint16Zint32Zint64strZ_tensor_classes __module____name__Z_storage_classesZ_utilsZ_rebuild_parameterZ_rebuild_tensorZ_rebuild_tensor_v2Z_rebuild_sparse_tensorZ_rebuild_meta_tensor_no_storageZ_tensorZ_rebuild_from_type_v2)rcttttsfr=\/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/_weights_only_unpickler.py_get_allowed_globalsDs>     r?c@s8eZdZddedddZddZddZd d Zd S) UnpicklerbytesencodingcCs ||_|j|_|j|_i|_dSN)rCreadlinereadmemo)selffilerCr=r=r>__init__rszUnpickler.__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"|dt7dkr|t4d|ddq"|dt8dkrt4d|dd}|t9krt dt:||dd}||q"|dt;dkr$|dd}||}|jkrnt dt|t|t'krt$|dkrtj?@|ddkrt d|d||A|q"|dtBdtCdfkr|dtBdkr|dn t4d|dd}||jD|q"|dtEdtFdfkr|dtEdkrP|dn t4d|dd}|dkrttGd |jd|jD|<q"|dtHdkr|dd}||}|tI|q"|dtJdkr|ddq"|dtKdkr|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=zdzloadxs$                     "*                ,,   zUnpickler.loadcCs |j}|j|_|jj|_|SrD)rTrSr[rU)rHrwr=r=r>rcs  zUnpickler.pop_markcCs tddS)Nz%unsupported persistent id encountered)r')rHr|r=r=r>rj!szUnpickler.persistent_loadN)r7r6 __qualname__r5rJrrcrjr=r=r=r>r@qs $r@ASCIIrBcCst||dS)NrB)r@r)rIrCr=r=r>r%sr)8 functools _functools collectionsrpicklerrrrrrr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'structr(sysr)typingr*r+r,r/ lru_cacher?r@r5rr=r=r=r>s '    ,5