diffusers
d773308c - Reduce WanAnimate TorchAO test input sizes to prevent OOM (#13541)

Commit
113 days ago
Reduce WanAnimate TorchAO test input sizes to prevent OOM (#13541) Shrink dummy inputs to avoid OOM on devices without FlashAttention. Reduce hidden_states spatial from 64x64 to 16x16 and frames from 21 to 5, bringing self-attention sequence length from 21,504 to 320. Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com> Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Author
Parents
Loading