Merge branch 'master' into dev

2026-04-27 17:51:22 +00:00 · 2023-05-02 09:25:47 +03:00
parent b1717c0a48 5ab7f213be
commit 4499bead4c
2 changed files with 11 additions and 3 deletions
--- a/modules/safe.py
+++ b/modules/safe.py
@@ -24,7 +24,11 @@ class RestrictedUnpickler(pickle.Unpickler):

    def persistent_load(self, saved_id):
        assert saved_id[0] == 'storage'
-        return TypedStorage(_internal=True)
+
+        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: