mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-27 04:11:39 -08:00
Code Issues Projects Releases Packages Wiki Activity

make launch.py run installers for extensions that have ones

Browse source 
add some more classes to safety module for an extension
This commit is contained in:
AUTOMATIC 2022-11-01 14:19:24 +03:00
parent f126986b76
commit d35bf64945
2 changed files with 21 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/safe.py
View file

@ -32,7 +32,7 @@ class RestrictedUnpickler(pickle.Unpickler):
return getattr(collections, name)
if module == 'torch._utils' and name in ['_rebuild_tensor_v2', '_rebuild_parameter']:
return getattr(torch._utils, name)
if module == 'torch' and name in ['FloatStorage', 'HalfStorage', 'IntStorage', 'LongStorage', 'DoubleStorage']:
if module == 'torch' and name in ['FloatStorage', 'HalfStorage', 'IntStorage', 'LongStorage', 'DoubleStorage', 'ByteStorage']:
return getattr(torch, name)
if module == 'torch.nn.modules.container' and name in ['ParameterDict']:
return getattr(torch.nn.modules.container, name)