Skip to content

Commit c1a68d7

Browse files
committed
linting
Signed-off-by: Antoni Viros i Martin <[email protected]>
1 parent 6c15512 commit c1a68d7

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

fms_mo/aiu_addons/fp8/fp8_attn.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -317,14 +317,16 @@ def _spyre_scaled_paged_compute_op(
317317
attn_kwargs["left_padded_prompt_mask"],
318318
attn_kwargs["block_table"],
319319
)
320-
320+
321321
def __spyre_scaled_paged_validate_attn_kwargs_op(
322322
input_ids: torch.Tensor,
323323
position_ids: torch.Tensor,
324324
past_key_value_states: Optional[list[tuple[torch.Tensor, torch.Tensor]]] = None,
325325
**attn_kwargs,
326326
):
327-
__spyre_paged_validate_attn_kwargs_op(input_ids, position_ids, past_key_value_states, **attn_kwargs)
327+
__spyre_paged_validate_attn_kwargs_op(
328+
input_ids, position_ids, past_key_value_states, **attn_kwargs
329+
)
328330

329331
if past_key_value_states is not None:
330332
for k, v in past_key_value_states:

0 commit comments

Comments
 (0)