diffusers
add reshape to fix use_memory_efficient_attention in flax
#7918

Merged

add reshape to fix use_memory_efficient_attention in flax #7918

pcuenca merged 5 commits into huggingface:main from entrpn:fix_flax_use_memory_efficient_attention

add reshape to fix use_memory_efficient_attention in flax

59595187

yiyixuxu requested a review from

pcuenca 1 year ago

Merge branch 'main' into fix_flax_use_memory_efficient_attention

5c9b9710

Merge branch 'main' into fix_flax_use_memory_efficient_attention

d10fa094

Merge branch 'main' into fix_flax_use_memory_efficient_attention

c3b1e2fd

github-actions added stale

a-r-r-o-w removed stale

Merge branch 'main' into fix_flax_use_memory_efficient_attention

216cddc4

github-actions added stale

pcuenca approved these changes on 2024-12-14

pcuenca merged a5f35ee4 into main 1 year ago

entrpn deleted the fix_flax_use_memory_efficient_attention branch 1 year ago

Reviewers

pcuenca

Assignees

No one assigned

Labels

stale

Milestone

No milestone