Skip to content

Commit d0646ad

Browse files
committed
Fix format for a file
1 parent e0d4d44 commit d0646ad

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

benchmark/bench_flash_attn.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -127,9 +127,7 @@ def benchmark(
127127
max_seqlen_q = q_seq_length
128128
window_size = (-1, -1) if not local else torch.randint(0, kv_seq_length, (2,))
129129

130-
sinks = (
131-
torch.randn(num_heads, device=device, dtype=dtype) if use_sinks else None
132-
)
130+
sinks = torch.randn(num_heads, device=device, dtype=dtype) if use_sinks else None
133131

134132
softmax_scale = 1.0 / (head_dim**0.5)
135133

0 commit comments

Comments
 (0)