Skip to content

Commit 0ea904e

Browse files
committed
add todos for future
1 parent d40bced commit 0ea904e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/diffusers/pipelines/pyramid_attention_broadcast_utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,9 @@ class PyramidAttentionBroadcastConfig:
8080
temporal_attention_block_identifiers: Tuple[str, ...] = _TEMPORAL_ATTENTION_BLOCK_IDENTIFIERS
8181
cross_attention_block_identifiers: Tuple[str, ...] = _CROSS_ATTENTION_BLOCK_IDENTIFIERS
8282

83+
# TODO(aryan): add PAB for MLP layers (very limited speedup from testing with original codebase
84+
# so not added for now)
85+
8386

8487
class PyramidAttentionBroadcastState:
8588
r"""

0 commit comments

Comments
 (0)