Update forge_reference.py

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