Commit graph

7 commits

Author SHA1 Message Date
AUTOMATIC
6e4de5b442 add load_with_extra function for modules to load checkpoints with extended whitelist 2022-11-06 11:20:23 +03:00
AUTOMATIC
d35bf64945 make launch.py run installers for extensions that have ones
add some more classes to safety module for an extension
2022-11-01 14:20:15 +03:00
AUTOMATIC
9e5ca5077f extra message for unpicking fails 2022-10-14 16:37:36 +03:00
AUTOMATIC
66b7d7584f become even stricter with pickles
no pickle shall pass
thank you again, RyotaK
2022-10-11 17:03:16 +03:00
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.
2022-10-10 08:04:52 +03:00
AUTOMATIC
a65476718f add DoubleStorage to list of allowed classes for pickle 2022-10-09 23:38:49 +03:00
AUTOMATIC
875ddfeecf added guard for torch.load to prevent loading pickles with unknown content 2022-10-09 17:58:43 +03:00