Commit 5ab7f213 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix an error that prevents running webui on torch<2.0 without --disable-safe-unpickle

parent 72cd27a1
Loading
Loading
Loading
Loading
+4 −0
Original line number Diff line number Diff line
## 1.1.1
### Bug Fixes:
 * fix an error that prevents running webui on torch<2.0 without --disable-safe-unpickle

## 1.1.0
### Features:
 * switch to torch 2.0.0 (except for AMD GPUs)
+5 −1
Original line number Diff line number Diff line
@@ -24,7 +24,11 @@ class RestrictedUnpickler(pickle.Unpickler):

    def persistent_load(self, saved_id):
        assert saved_id[0] == 'storage'

        try:
            return TypedStorage(_internal=True)
        except TypeError:
            return TypedStorage()  # PyTorch before 2.0 does not have the _internal argument

    def find_class(self, module, name):
        if self.extra_handler is not None: