SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[SDPA] Make sure attn mask creation is always done on CPU (#28400)

Browse Source 
* [SDPA] Make sure attn mask creation is always done on CPU

* Update docker to 2.1.1

* revert test change
This commit is contained in:
Patrick von Platen
2024-01-09 11:05:19 +01:00
committed by GitHub
parent 5c7e11e010
commit 8604dd308d
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docker/transformers-all-latest-gpu/Dockerfile
View File

@@ -9,9 +9,9 @@ SHELL ["sh", "-lc"]
# The following `ARG` are mainly used to specify the versions explicitly & directly in this docker file, and not meant
# to be used as arguments for docker build (so far).
ARG PYTORCH='2.1.0'
ARG PYTORCH='2.1.1'
# (not always a valid torch version)
ARG INTEL_TORCH_EXT='2.1.0'
ARG INTEL_TORCH_EXT='2.1.1'
# Example: `cu102`, `cu113`, etc.
ARG CUDA='cu118'