Skip to content

Commit 56b1837

Browse files
committed
tests/attention: black format fp8_13 ModelConfig
Line was 105 chars; black requires <=100 with the project's preview+ string_processing settings. Signed-off-by: Vedaanta Agarwalla <vagarwalla@nvidia.com>
1 parent 6b4720a commit 56b1837

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

tests/pytorch/attention/test_attention.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1928,7 +1928,9 @@ def get_model(dtype, config):
19281928
attn_mask_type="causal_bottom_right",
19291929
),
19301930
"fp8_12": ModelConfig(1, 8192, 32, 128, num_gqa_groups=4, attn_mask_type="causal"),
1931-
"fp8_13": ModelConfig(2, 8192, 32, 128, num_gqa_groups=4, attn_mask_type="causal", window_size=(128, 0)),
1931+
"fp8_13": ModelConfig(
1932+
2, 8192, 32, 128, num_gqa_groups=4, attn_mask_type="causal", window_size=(128, 0)
1933+
),
19321934
"fp8_14": ModelConfig(2, 4096, 64, 64, num_gqa_groups=8, attn_mask_type="causal"),
19331935
"fp8_15": ModelConfig(1, 8192, 64, 64, attn_mask_type="causal", window_size=(128, 0)),
19341936
"fp8_16": ModelConfig(

0 commit comments

Comments
 (0)