[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
76376d37f0
commit
2eda89058c
1 changed files with 3 additions and 1 deletions
|
|
@ -85,7 +85,9 @@ class AttentionContext:
|
|||
sliding_window: Optional[int] = None
|
||||
|
||||
|
||||
def select_attention_backend(use_varlen: bool = False, has_attention_mask: bool = False) -> str:
|
||||
def select_attention_backend(
|
||||
use_varlen: bool = False, has_attention_mask: bool = False
|
||||
) -> str:
|
||||
"""Return attention backend based on availability / priority order."""
|
||||
|
||||
if has_attention_mask:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue