reverts accidental change that removes attn_mask in attn. Improves fl…#11065

Merged

yiyixuxu merged 1 commit intohuggingface:mainfrom

entrpn:update_ptxla_flux

Mar 14, 2025

Commits on Mar 14, 2025

reverts accidental change that removes attn_mask in attn. Improves flux ptxla by using flash block sizes. Moves encoding outside the for loop.
jfacevedo-google
committed