SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
07d79520efd7fc42ac9ce0a055f7ea6e583bc3a8
HuggingFace_transformer/docker/transformers-pytorch-gpu
History
Patrick von Platen 8604dd308d [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
2024-01-09 11:05:19 +01:00
..
Dockerfile
[SDPA] Make sure attn mask creation is always done on CPU (#28400)
2024-01-09 11:05:19 +01:00