aboutsummaryrefslogtreecommitdiffstats
path: root/modules/safe.py
AgeCommit message (Collapse)AuthorLines
2022-12-02Fixed safe.py for pytorch 1.13 ckpt filesSmirkingFace-7/+11
2022-11-06add load_with_extra function for modules to load checkpoints with extended ↵AUTOMATIC-3/+37
whitelist
2022-11-01make launch.py run installers for extensions that have onesAUTOMATIC-1/+1
add some more classes to safety module for an extension
2022-10-14extra message for unpicking failsAUTOMATIC-1/+8
2022-10-11become even stricter with picklesAUTOMATIC-0/+17
no pickle shall pass thank you again, RyotaK
2022-10-10Newer versions of PyTorch use TypedStorage insteadbrkirch-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.
2022-10-09add DoubleStorage to list of allowed classes for pickleAUTOMATIC-1/+1
2022-10-09added guard for torch.load to prevent loading pickles with unknown contentAUTOMATIC-0/+89