Commit Graph

19 Commits

Author SHA1 Message Date
catboxanon
cb5f61281a
Allow bf16 in safe unpickler 2023-05-13 11:04:26 -04:00
AUTOMATIC
a5121e7a06 fixes for B007 2023-05-10 11:37:18 +03:00
AUTOMATIC
550256db1c ruff manual fixes 2023-05-10 11:19:16 +03:00
AUTOMATIC
5ab7f213be fix an error that prevents running webui on torch<2.0 without --disable-safe-unpickle 2023-05-02 09:20:35 +03:00
AUTOMATIC
ee71eee181 stuff related to torch version change 2023-04-29 12:36:50 +03:00
AUTOMATIC1111
b12de850ae
Merge pull request #5992 from yuvalabou/F541
Fix F541: f-string without any placeholders
2022-12-25 09:16:08 +03:00
AUTOMATIC
8eef9d8e78 a way to add an exception to unpickler without explicitly calling load_with_extra 2022-12-25 09:03:56 +03:00
Yuval Aboulafia
3bf5591efe fix F541 f-string without any placeholders 2022-12-24 21:35:29 +02:00
AUTOMATIC1111
fac92610d2
Merge pull request #5753 from calvinballing/master
Fix various typos
2022-12-24 09:58:28 +03:00
brkirch
cca16373de Add attributes used by MPS 2022-12-17 04:23:08 -05:00
Jim Hays
c0355caefe Fix various typos 2022-12-14 21:01:32 -05:00
SmirkingFace
e461477869 Fixed safe.py for pytorch 1.13 ckpt files 2022-12-02 11:12:13 +01:00
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