Skip to content

Commit 610b0fe

Browse files
committed
Wrong tensor for comparison
1 parent 4d571ed commit 610b0fe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/model-conversion/scripts/causal/run-org-model-multi-token.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def summarize(tensor: torch.Tensor, name: str, max_seq: int = 3, max_vals: int =
123123
print(f" sum = {t.sum().item():.6f}\n")
124124

125125
indexed_patterns = [ r"model\.layers\.[0-9]+_out", r"recurrent_cache_[0-9]+" ]
126-
non_indexed_patterns = [ r"k_pad", r"v_pad", r"q_pad" ]
126+
non_indexed_patterns = [ r"k_pad", r"v_pad", r"q_scaled" ]
127127

128128
if any(re.fullmatch(p, name) for p in indexed_patterns):
129129
if name not in token_counter:

0 commit comments

Comments
 (0)