aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--webui.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/webui.py b/webui.py
index 49002a2a..ad38323e 100644
--- a/webui.py
+++ b/webui.py
@@ -730,6 +730,8 @@ class StableDiffusionModelHijack:
data = torch.load(path)
param_dict = data['string_to_param']
+ if hasattr(param_dict, '_parameters'):
+ param_dict = getattr(param_dict, '_parameters') # fix for torch 1.12.1 loading saved file from torch 1.11
assert len(param_dict) == 1, 'embedding file has multiple terms in it'
emb = next(iter(param_dict.items()))[1].reshape(768)
self.word_embeddings[name] = emb