diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-02 06:20:35 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-02 06:20:35 +0000 |
| commit | 5ab7f213bec2f816f9c5644becb32eb72c8ffb89 (patch) | |
| tree | 6795f99d10f7a93bafbf04b5f76148c2a0a491c7 | |
| parent | 72cd27a13587c9579942577e9e3880778be195f6 (diff) | |
| download | stable-diffusion-webui-gfx803-5ab7f213bec2f816f9c5644becb32eb72c8ffb89.tar.gz stable-diffusion-webui-gfx803-5ab7f213bec2f816f9c5644becb32eb72c8ffb89.tar.bz2 stable-diffusion-webui-gfx803-5ab7f213bec2f816f9c5644becb32eb72c8ffb89.zip | |
fix an error that prevents running webui on torch<2.0 without --disable-safe-unpickle
| -rw-r--r-- | CHANGELOG.md | 4 | ||||
| -rw-r--r-- | modules/safe.py | 6 |
2 files changed, 9 insertions, 1 deletions
diff --git a/CHANGELOG.md b/CHANGELOG.md index 8b8a3611..8d2f96e5 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,3 +1,7 @@ +## 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)
diff --git a/modules/safe.py b/modules/safe.py index dadf319c..e6c2f2c0 100644 --- 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:
|