a d)@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 complex128float16float32float64Zint8int16int32int64strZ_tensor_classes __module____name__Z_storage_classes_utilsZ_rebuild_parameterZ_rebuild_tensorZ_rebuild_tensor_v2Z_rebuild_sparse_tensorZ_rebuild_meta_tensor_no_storageZ_tensorZ_rebuild_from_type_v2)rcttttsfrDf/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/_weights_only_unpickler.py_get_allowed_globalsDs>     rFc@s8eZdZddedddZddZddZd d Zd S) UnpicklerbytesencodingcCs ||_|j|_|j|_i|_dSN)rJreadlinereadmemo)selffilerJrDrDrE__init__rszUnpickler.__init__cCsg|_g|_|jj|_|j}|j}|d}|s2tt|ts@J|dtdkr|dd d}|dd d}|d|}|t vr|t |nt d|q"|dt dkr|j }|j }|tjjurt d||tjj|q"|dtdkrZ|j }|jd} | t vrJt d | | ||jd<q"|dtdkr|j } |jd} t| tjur| j| nLt| tjjur| | n.t| tur| j| nt d t| q"|dtdkr<|j } |jd} t| tur0t d t| | | q"|dtdkr|}|jd} t| turt 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"|dt/dkr|iq"|dt0dkr|t1q"|dt2dkr|t3d|ddq"|dt4dkr,||ddq"|dt5dkrX|t3d|d dq"|dt6dkr|t3d|ddq"|dt7dkrt3d|dd}|t8krt dt9||dd}||q"|dt:dkr$|dd}||}|j;dkr| |j;d}||q"|dt?|ddkrt d|d||@|q"|dtAdtBdfvr|dtAdkr|dn t3d|dd}||jC|q"|dtDdtEdfvr|dtDdkrP|dn t3d|dd}|dkrttFd |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 FTrDzdzs '    ,5