diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 21:21:32 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 21:21:32 +0300 |
| commit | 3ec7b705c78b7aca9569c92a419837352c7a4ec6 (patch) | |
| tree | 98248bc21aa4ad9715205f0a65a654532c6cfcc0 /modules/textual_inversion/textual_inversion.py | |
| parent | d25219b7e889cf34bccae9cb88497708796efda2 (diff) | |
suggestions and fixes from the PR
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 47035332..9e1b2b9a 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -166,8 +166,7 @@ class EmbeddingDatabase: # textual inversion embeddings
if 'string_to_param' in data:
param_dict = data['string_to_param']
- if hasattr(param_dict, '_parameters'):
- param_dict = param_dict._parameters # fix for torch 1.12.1 loading saved file from torch 1.11
+ param_dict = getattr(param_dict, '_parameters', param_dict) # 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]
# diffuser concepts
|