diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-05-11 18:28:15 +0300 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-05-11 20:29:11 +0300 |
| commit | 49a55b410b66b7dd9be9335d8a2e3a71e4f8b15c (patch) | |
| tree | d79f004eae46bc1c49832f3c668a524107c30034 /modules/sub_quadratic_attention.py | |
| parent | 431bc5a297ff7c17231b92b6c8f8152b2fab8553 (diff) | |
Autofix Ruff W (not W605) (mostly whitespace)
Diffstat (limited to 'modules/sub_quadratic_attention.py')
| -rw-r--r-- | modules/sub_quadratic_attention.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sub_quadratic_attention.py b/modules/sub_quadratic_attention.py index cc38debd..497568eb 100644 --- a/modules/sub_quadratic_attention.py +++ b/modules/sub_quadratic_attention.py @@ -179,7 +179,7 @@ def efficient_dot_product_attention( chunk_idx, min(query_chunk_size, q_tokens) ) - + summarize_chunk: SummarizeChunk = partial(_summarize_chunk, scale=scale) summarize_chunk: SummarizeChunk = partial(checkpoint, summarize_chunk) if use_checkpoint else summarize_chunk compute_query_chunk_attn: ComputeQueryChunkAttn = partial( |