Skip to content

Commit 8b580f3

Browse files
committed
fix style
Signed-off-by: Max Jeblick <maximilianjeblick@gmail.com>
1 parent a1f30e1 commit 8b580f3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

kvpress/presses/qfilter_press.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ def __init__(self, num_layers: int, num_kv_heads: int, kv_head_dim: int):
1616
super().__init__()
1717
self.q_filters = torch.nn.Parameter(torch.randn(num_layers, num_kv_heads, kv_head_dim))
1818

19+
1920
@dataclass
2021
class QFilterPress(ScorerPress):
2122
"""

0 commit comments

Comments
 (0)