mirror of
https://github.com/vladmandic/automatic
synced 2026-09-12 07:58:43 +02:00
update attention handlers and settings
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -546,9 +546,8 @@ def update_pipeline(sd_model, p: processing.StableDiffusionProcessing):
|
||||
global orig_pipeline # pylint: disable=global-statement
|
||||
orig_pipeline = updated_model # processed ONNX pipeline should not be replaced with original pipeline.
|
||||
current_attn = getattr(updated_model, "current_attn_name", None)
|
||||
current_overrides = getattr(updated_model, "current_attn_overrides", None)
|
||||
if current_attn != shared.opts.cross_attention_optimization or current_overrides != list(shared.opts.sdp_overrides):
|
||||
log.info(f"Setting attention optimization: {shared.opts.cross_attention_optimization} overrides={shared.opts.sdp_overrides}")
|
||||
if current_attn != shared.opts.cross_attention_optimization:
|
||||
# log.info(f"Setting attention optimization: {shared.opts.cross_attention_optimization}")
|
||||
attention.set_diffusers_attention(updated_model)
|
||||
return updated_model
|
||||
|
||||
|
||||
Reference in New Issue
Block a user