Update forge_reference.py

2026-02-01 22:09:46 +00:00 · 2024-01-30 19:39:04 -08:00
parent 2b58f91505
commit 63ea205bed
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
@@ -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