From d127935eadbd8ec351a52eb12f37527d4579b071 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Tue, 30 Jan 2024 18:25:50 -0800 Subject: [PATCH] Update forge_reference.py --- .../scripts/forge_reference.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py b/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py index 2fc6743e..379dfa29 100644 --- a/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py +++ b/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py @@ -56,9 +56,19 @@ class PreprocessorReference(Preprocessor): return q, k, v + def attn1_output_proc(h, transformer_options): + sigma = transformer_options["sigmas"][0].item() + if not (sigma_min <= sigma <= sigma_max): + return h + + a = 0 + + return h + unet.add_block_modifier(block_proc) unet.add_conditioning_modifier(conditioning_modifier) unet.set_model_attn1_patch(attn1_proc) + unet.set_model_attn1_patch(attn1_output_proc) process.sd_model.forge_objects.unet = unet