fix pulid attention

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-11-14 17:08:11 -05:00
parent c8b08f49cb
commit 7344a8c3d8
2 changed files with 3 additions and 2 deletions
+2 -2
View File
@@ -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
+1
View File
@@ -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}')