aboutsummaryrefslogtreecommitdiffstats
path: root/modules/safe.py
Commit message (Collapse)AuthorAgeFilesLines
* add load_with_extra function for modules to load checkpoints with extended ↵AUTOMATIC2022-11-061-3/+37
| | | | whitelist
* make launch.py run installers for extensions that have onesAUTOMATIC2022-11-011-1/+1
| | | | add some more classes to safety module for an extension
* extra message for unpicking failsAUTOMATIC2022-10-141-1/+8
|
* become even stricter with picklesAUTOMATIC2022-10-111-0/+17
| | | | | no pickle shall pass thank you again, RyotaK
* Newer versions of PyTorch use TypedStorage insteadbrkirch2022-10-101-1/+5
| | | | Pytorch 1.13 and later will rename _TypedStorage to TypedStorage, so check for TypedStorage and use _TypedStorage if it is not available. Currently this is needed so that nightly builds of PyTorch work correctly.
* add DoubleStorage to list of allowed classes for pickleAUTOMATIC2022-10-091-1/+1
|
* added guard for torch.load to prevent loading pickles with unknown contentAUTOMATIC2022-10-091-0/+89