diff --git a/modules/sd_models.py b/modules/sd_models.py index be543cb49..e34f7c6f9 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -1188,8 +1188,8 @@ def set_diffusers_attention(pipe): else: module.set_attn_processor(attention) - if hasattr(pipe, 'pipe'): - set_diffusers_attention(pipe.pipe) + # if hasattr(pipe, 'pipe'): + # set_diffusers_attention(pipe.pipe) if 'ControlNet' in pipe.__class__.__name__: # do not replace attention in ControlNet pipelines return diff --git a/scripts/pulid_ext.py b/scripts/pulid_ext.py index d01ca2847..40a21e3f7 100644 --- a/scripts/pulid_ext.py +++ b/scripts/pulid_ext.py @@ -178,6 +178,7 @@ class Script(scripts.Script): sd_models.copy_diffuser_options(shared.sd_model, shared.sd_model.pipe) sd_models.move_model(shared.sd_model, devices.device) # move pipeline to device sd_models.set_diffuser_options(shared.sd_model, vae=None, op='model') + # shared.sd_model.hack_unet_attn_layers(shared.sd_model.pipe.unet) # reapply attention layers devices.torch_gc() except Exception as e: shared.log.error(f'PuLID: failed to create pipeline: {e}')