transformers
8604dd30
- [SDPA] Make sure attn mask creation is always done on CPU (#28400)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
[SDPA] Make sure attn mask creation is always done on CPU (#28400) * [SDPA] Make sure attn mask creation is always done on CPU * Update docker to 2.1.1 * revert test change
References
#28400 - [SDPA] Make sure attn mask creation is always done on CPU
Author
patrickvonplaten
Parents
5c7e11e0
Loading