diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-05-11 21:25:15 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-11 21:25:15 +0300 |
| commit | abe32cefa39dee36d7f661d4e63c28ea8dd60c4f (patch) | |
| tree | 1f1d817b59b49c6d3944c959151ce4c67d9041da /modules/sub_quadratic_attention.py | |
| parent | b4aaa339d529c81859858f0bedcc72b44fccd3d0 (diff) | |
| parent | 49a55b410b66b7dd9be9335d8a2e3a71e4f8b15c (diff) | |
Merge pull request #10285 from akx/ruff-spacing
Indentation + ruff whitespace fixes
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( |