Skip to content

add reshape to fix use_memory_efficient_attention in flax - #7918

Merged
pcuenca merged 5 commits into
huggingface:mainfrom
entrpn:fix_flax_use_memory_efficient_attention
Dec 14, 2024
Merged

add reshape to fix use_memory_efficient_attention in flax#7918
pcuenca merged 5 commits into
huggingface:mainfrom
entrpn:fix_flax_use_memory_efficient_attention

Merge branch 'main' into fix_flax_use_memory_efficient_attention

216cddc
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs