a Ad@sddlZddlZddlZddlZddlZddlZddlZddlmZe ej drXej j nej j Z ddZ GdddejZedZed Zd d Zd d ZddZdddZGdddZejZee_dadS)N)errors TypedStoragecGstj|}|SN)_codecsencode)argsoutr 4/var/www/html/stable-diffusion-webui/modules/safe.pyrs rc@s eZdZdZddZddZdS)RestrictedUnpicklerNcCs:|ddksJz tddWSty4tYS0dS)NrstorageT) _internal)r TypeError)selfsaved_idr r r persistent_loads   z#RestrictedUnpickler.persistent_loadcCsV|jdur"|||}|dur"|S|dkr<|dkrWzcheck_pt..rzdata.pkl not found in zMultiple data.pkl found in rb) zipfileZipFilerCnamelistlenr6openr r+load BadZipfilerange)rAr+zZdata_pkl_filenamesfile unpickler_r r r check_ptOs$   H  rXcOst|g|Rdti|S)Nr+)load_with_extraglobal_extra_handler)rArkwargsr r r rQksrQcOsddlm}z|jjs t||WnTtjyNtjd|dddYdSt yvtjd|dddYdS0t |g|Ri|S) a this function is intended to be used by extensions that want to load models with some extra classes in them that the usual unpickler would find suspicious. Use the extra_handler argument to specify a function that takes module and field name as text, and returns that field's value: ```python def extra(module, name): if module == 'collections' and name == 'OrderedDict': return collections.OrderedDict return None safe.load_with_extra('model.pt', extra_handler=extra) ``` The alternative to this is just to use safe.unsafe_torch_load('model.pt'), which as the name implies is definitely unsafe. r)sharedz"Error verifying pickled file from z -----> !!!! The file is most likely corrupted !!!! <----- You can skip this check with --disable-safe-unpickle commandline argument, but that is not going to help you. T)exc_infoNz The file may be malicious, so the program is not going to read it. You can skip this check with --disable-safe-unpickle commandline argument. ) r/r\cmd_optsdisable_safe_unpicklerXpickleUnpicklingErrorrreportr6unsafe_torch_load)rAr+rr[r\r r r rYos"    rYc@s(eZdZdZddZddZddZdS) Extraa A class for temporarily setting the global handler for when you can't explicitly call load_with_extra (because it's not your code making the torch.load call). The intended use is like this: ``` import torch from modules import safe def handler(module, name): if module == 'torch' and name in ['float64', 'float16']: return getattr(torch, name) return None with safe.Extra(handler): x = torch.load('model.pt') ``` cCs ||_dSr)handler)rrer r r __init__szExtra.__init__cCstdusJd|jadS)Nzalready inside an Extra() block)rZre)rr r r __enter__szExtra.__enter__cCsdadSr)rZ)rexc_typeexc_valexc_tbr r r __exit__szExtra.__exit__N)r<r=r>__doc__rfrgrkr r r r rdsrd)N)r`rrr#rrLrer/rhasattrr rZ _TypedStorager Unpicklerr compiler?rDrCrXrQrYrdrcrZr r r r s( -   0#