mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
@@ -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
|
||||
|
||||
@@ -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}')
|
||||
|
||||
Reference in New Issue
Block a user