Update forge_reference.py

2026-02-04 23:19:57 +00:00 · 2024-01-30 18:09:19 -08:00
parent c0dc02c488
commit 28b7af44cc
1 changed files with 2 additions and 2 deletions
--- 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)