SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
cbe58b4269457a6ca66a556224b23f9ef246f905
HuggingFace_transformer/docker/transformers-pytorch-gpu/Dockerfile
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

1.6 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink