mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-30 04:11:15 -08:00
Code Issues Projects Releases Packages Wiki Activity

added guard for torch.load to prevent loading pickles with unknown content

Browse source
This commit is contained in:
AUTOMATIC 2022-10-09 17:58:43 +03:00
parent bba2ac8324
commit 875ddfeecf
3 changed files with 91 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
modules/paths.py
View file

@ -1,6 +1,7 @@
import argparse
import os
import sys
import modules.safe
script_path = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
models_path = os.path.join(script_path, "models")