add a possible fix for 'LatentDiffusion' object has no attribute 'lora_layer_mapping'

This commit is contained in:
AUTOMATIC
2023-05-14 13:31:03 +03:00
parent d7e9ac2aff
commit b9abdb50a3
2 changed files with 5 additions and 0 deletions

View File

@@ -133,6 +133,10 @@ def load_lora(name, filename):
sd = sd_models.read_state_dict(filename)
# this should not be needed but is here as an emergency fix for an unknown error people are experiencing in 1.2.0
if not hasattr(shared.sd_model, 'lora_layer_mapping'):
assign_lora_names_to_compvis_modules(shared.sd_model)
keys_failed_to_match = {}
is_sd2 = 'model_transformer_resblocks' in shared.sd_model.lora_layer_mapping