From 28b7af44cc31969cff10f5c8a68aa086b11ec828 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Tue, 30 Jan 2024 18:09:19 -0800 Subject: [PATCH] Update forge_reference.py --- .../forge_preprocessor_reference/scripts/forge_reference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py b/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py index a247c966..30719f30 100644 --- a/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py +++ b/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py @@ -50,11 +50,11 @@ class PreprocessorReference(Preprocessor): 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 + return q, k, v a = 0 - return q, k, k + return q, k, v unet.add_block_modifier(block_proc) unet.add_conditioning_modifier(conditioning_modifier)