From c0dc02c488cf469b8d75cfe0ee9daf98094673fb Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Tue, 30 Jan 2024 18:08:15 -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 f2a78661..a247c966 100644 --- a/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py +++ b/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py @@ -47,8 +47,18 @@ class PreprocessorReference(Preprocessor): return h + def attn1_proc(q, k, v, transformer_options): + sigma = transformer_options["sigmas"][0].item() + if not (sigma_min < sigma < sigma_max): + return q, k, k + + a = 0 + + return q, k, k + unet.add_block_modifier(block_proc) unet.add_conditioning_modifier(conditioning_modifier) + unet.set_model_attn1_patch(attn1_proc) process.sd_model.forge_objects.unet = unet