9 Commits (c0355caefe3d82e304e6d832699d581fc8f9fbf9)

Author SHA1 Message Date
Jim Hays c0355caefe Fix various typos 3 years ago
SmirkingFace e461477869 Fixed safe.py for pytorch 1.13 ckpt files 3 years ago
AUTOMATIC 6e4de5b442 add load_with_extra function for modules to load checkpoints with extended whitelist 3 years ago
AUTOMATIC d35bf64945 make launch.py run installers for extensions that have ones
add some more classes to safety module for an extension
3 years ago
AUTOMATIC 9e5ca5077f extra message for unpicking fails 3 years ago
AUTOMATIC 66b7d7584f become even stricter with pickles
no pickle shall pass
thank you again, RyotaK
3 years ago
brkirch 8acc901ba3 Newer versions of PyTorch use TypedStorage instead
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.
3 years ago
AUTOMATIC a65476718f add DoubleStorage to list of allowed classes for pickle 3 years ago
AUTOMATIC 875ddfeecf added guard for torch.load to prevent loading pickles with unknown content 3 years ago