From 63ea205bed560c0646d7ca10e3249cf9735e50da Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Tue, 30 Jan 2024 19:39:04 -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 0461ea1b..f5adce62 100644 --- a/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py +++ b/extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py @@ -79,7 +79,7 @@ class PreprocessorReference(Preprocessor): return h if self.is_recording_style: - a = 0 + self.recorded_h[location] = h else: b = 0 @@ -97,7 +97,7 @@ class PreprocessorReference(Preprocessor): transformer_options['block_index']) if self.is_recording_style: - a = 0 + self.recorded_attn1[location] = (k, v) else: b = 0