Commit 6b9a333
reverts accidental change that removes attn_mask in attn. Improves fl… (#11065)
reverts accidental change that removes attn_mask in attn. Improves flux ptxla by using flash block sizes. Moves encoding outside the for loop.
Co-authored-by: Juan Acevedo <jfacevedo@google.com>1 parent 8ead643 commit 6b9a333
3 files changed
Lines changed: 133 additions & 52 deletions
File tree
- examples/research_projects/pytorch_xla/inference/flux
- src/diffusers/models
0 commit comments