From 7bc42fe9f6c2f9859e3c83cc7a9714798be648ff Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Tue, 30 Jan 2024 19:46:45 -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 bb787e17..288352ad 100644 --- a/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py +++ b/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py @@ -82,7 +82,7 @@ class PreprocessorReference(Preprocessor): self.recorded_h[location] = h else: cond_mark = transformer_options['cond_mark'][:, None, None, None] # cond is 0 - recorded_h = self.recorded_h[location] + rh = self.recorded_h[location] b = 0 return h @@ -102,7 +102,7 @@ class PreprocessorReference(Preprocessor): self.recorded_attn1[location] = (k, v) else: cond_mark = transformer_options['cond_mark'][:, None, None, None] # cond is 0 - recorded_attn1 = self.recorded_attn1[location] + rk, rv = self.recorded_attn1[location] b = 0 return sdp(q, k, v, transformer_options)