transformers
ccc0381d - Fix flex_attention in training mode (#35605)

Commit

346 days ago

Fix flex_attention in training mode (#35605) * fix flex * add test * style

References

#35605 - Fix flex_attention in training mode

Author

Cyrilvallez

Cyrilvallez

Parents

Loading