diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-22 05:20:48 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-22 05:20:48 +0000 |
| commit | 0792fae078ba362a5119f56d84e3f490a88690ae (patch) | |
| tree | 599ca17b2f535fe20c37a793fc673e58188e53cb /modules/sd_disable_initialization.py | |
| parent | 2621566153920eb70bfa439f3d7c126ee8d36ec8 (diff) | |
| download | stable-diffusion-webui-gfx803-0792fae078ba362a5119f56d84e3f490a88690ae.tar.gz stable-diffusion-webui-gfx803-0792fae078ba362a5119f56d84e3f490a88690ae.tar.bz2 stable-diffusion-webui-gfx803-0792fae078ba362a5119f56d84e3f490a88690ae.zip | |
fix missing field for aesthetic embedding extension
Diffstat (limited to 'modules/sd_disable_initialization.py')
| -rw-r--r-- | modules/sd_disable_initialization.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/sd_disable_initialization.py b/modules/sd_disable_initialization.py index c72d8efc..e90aa9fe 100644 --- a/modules/sd_disable_initialization.py +++ b/modules/sd_disable_initialization.py @@ -41,7 +41,9 @@ class DisableInitialization: return self.create_model_and_transforms(*args, pretrained=None, **kwargs)
def CLIPTextModel_from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs):
- return self.CLIPTextModel_from_pretrained(None, *model_args, config=pretrained_model_name_or_path, state_dict={}, **kwargs)
+ res = self.CLIPTextModel_from_pretrained(None, *model_args, config=pretrained_model_name_or_path, state_dict={}, **kwargs)
+ res.name_or_path = pretrained_model_name_or_path
+ return res
def transformers_modeling_utils_load_pretrained_model(*args, **kwargs):
args = args[0:3] + ('/', ) + args[4:] # resolved_archive_file; must set it to something to prevent what seems to be a bug
|