Skip to content

Commit a1a8947

Browse files
authored
Merge pull request #196 from jasonlim131/patch-2
Fix mapping typo in modelings_intervenable_gemma2.py
2 parents 8b071ed + 3a6c448 commit a1a8947

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pyvene/models/gemma2/modelings_intervenable_gemma2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
"value_output": ("hidden_size",),
5050
"head_query_output": ("head_dim",),
5151
"head_key_output": ("head_dim",),
52-
"head_value_output": ("hhead_dim",),
52+
"head_value_output": ("head_dim",),
5353
}
5454

5555

0 commit comments

Comments
 (0)