diff options
author | InvincibleDude <81354513+InvincibleDude@users.noreply.github.com> | 2023-01-24 15:44:09 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-01-24 15:44:09 +0300 |
commit | 44c0e6b993d00bb2f441f0fde409bcb79136f034 (patch) | |
tree | e27a45d1a3ceb8aab884631c7a806c5fe2c8386d /modules/sd_disable_initialization.py | |
parent | 3bc8ee998db5f461b8011a72e6f167012ccb8bc1 (diff) | |
parent | 602a1864b05075ca4283986e6f5c7d5bce864e11 (diff) |
Merge branch 'AUTOMATIC1111:master' into master
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
|