Flax memory efficient attention #2889
patrickvonplaten
merged
24 commits into
main
from
2231-flax-memory-efficient-attention
add use_memory_efficient params placeholder
61848154
test
e2d5708c
Merge branch 'huggingface:main' into main
c2221d6c
add memory efficient attention jax
9347cc58
Merge branch 'huggingface:main' into main
31d96a04
Merge branch 'main' of github.com:MuhHanif/diffusers into main
cb4c8ab9
add memory efficient attention jax
eac25e0d
newline
99d88e63
forgot dot
22557946
Merge branch 'huggingface:main' into main
f40df28b
Merge branch 'huggingface:main' into main
a4ff0b76
Rename use_memory_efficient
00803e94
Merge branch 'main' into 2231
95603712
Keep dtype last.
26ba0c44
Actually use key_chunk_size
4b82e44e
Rename symbol
78a106e4
Apply style
41ea7c28
Rename use_memory_efficient
f32f3313
Keep dtype last
00be593f
Pass `use_memory_efficient_attention` in `from_pretrained`
b5544f23
pcuenca
commented
on 2023-03-29
patrickvonplaten
commented
on 2023-03-30
patrickvonplaten
commented
on 2023-03-30
patrickvonplaten
commented
on 2023-03-30
patrickvonplaten
approved these changes
on 2023-03-30
Move JAX memory efficient attention to attention_flax.
8d35f092
yiyixuxu
approved these changes
on 2023-04-11
Simple test.
952293c1
style
d1490bcd
Merge remote-tracking branch 'origin/main' into 2231-flax-memory-effi…
ed003010
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub