[core] Pyramid Attention Broadcast #9562
start pyramid attention broadcast
67c729d4
add coauthor
6d3bdb55
update
37371016
make style
d5c738de
Merge branch 'main' into pyramid-attention-broadcast
1c97e04f
update
ae4abb14
Merge branch 'main' into pyramid-attention-broadcast
955e4f7c
make style
9f6987fd
a-r-r-o-w
marked this pull request as ready for review 1 year ago
add docs
b3547c64
add tests
afd0c176
update
6265b654
stevhliu
approved these changes
on 2024-10-07
Merge branch 'main' into pyramid-attention-broadcast
6816fe15
Update docs/source/en/api/pipelines/cogvideox.md
9cb4e876
Update docs/source/en/api/pipelines/cogvideox.md
6b1f55ec
Merge branch 'main' into pyramid-attention-broadcast
37d23669
Merge branch 'main' into pyramid-attention-broadcast
a5f51bba
Merge branch 'main' into pyramid-attention-broadcast
18b7d6d9
Pyramid Attention Broadcast rewrite + introduce hooks (#9826)
c52cf422
Merge branch 'main' into pyramid-attention-broadcast
3de2c189
Merge branch 'main' into pyramid-attention-broadcast
d95d61ae
merge pyramid-attention-rewrite-2
60905752
Merge branch 'main' into pyramid-attention-broadcast
af51f5dd
a-r-r-o-w
marked this pull request as draft 303 days ago
make style
903514ff
remove changes from latte transformer
b690db29
revert docs changes
63ab886a
better debug message
d40bceda
add todos for future
0ea904ec
update tests
9d452dc4
make style
cfe39216
cleanup
b972c4b1
fix
2b558ffa
improve log message; fix latte test
0b2629db
refactor
9182f57c
Merge branch 'main' into pyramid-attention-broadcast
d9744012
update
62b5b8dd
update
bb250d60
update
cbc086f1
revert changes to tests
7debcec2
update docs
a5c34afc
Merge branch 'main' into pyramid-attention-broadcast
ad242696
a-r-r-o-w
marked this pull request as ready for review 282 days ago
update tests
bbcde6b0
Apply suggestions from code review
b148ab4b
update
d4ecd6c9
fix flux test
6cca58fd
Merge branch 'main' into pyramid-attention-broadcast
c2e0e3b4
reorder
d9fad00f
Merge branch 'main' into pyramid-attention-broadcast
35296eb9
refactor
2436b3fb
make fix-copies
95c81482
update docs
76afc6a9
fixes
fb661678
more fixes
1040c911
make style
ffbabb59
update tests
1b92b1dd
update code example
88d917dc
make fix-copies
e4d8b12d
Merge branch 'main' into pyramid-attention-broadcast
cc946472
DN6
commented
on 2025-01-15
Merge branch 'main' into pyramid-attention-broadcast
071a0bad
refactor based on reviews
ae8bd994
use maybe_free_model_hooks
a9ee5a42
CacheMixin
1a596883
make style
c8616a6d
update
08a209d1
add current_timestep property; update docs
15e645df
make fix-copies
d6ce4ab1
update
96fce86f
improve tests
107e3752
Merge branch 'main' into pyramid-attention-broadcast
f7d7e387
try circular import fix
40fc7a50
apply suggestions from review
248f1039
DN6
commented
on 2025-01-16
Merge branch 'main' into pyramid-attention-broadcast
0a290a62
address review comments
fe939754
Apply suggestions from code review
2b599941
DN6
approved these changes
on 2025-01-21
Merge branch 'main' into pyramid-attention-broadcast
a8e460ef
refactor hook implementation
8c74a7ac
add test suite for hooks
3f3e26a3
PAB Refactor (#10667)
83d221f7
Merge branch 'main' into pyramid-attention-broadcast
847760ee
update
3d269ad8
fix remove hook behaviour
5535fd69
a-r-r-o-w
merged
658e24e8
into main 254 days ago
a-r-r-o-w
deleted the pyramid-attention-broadcast branch 254 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub