riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 13:20:12 +06:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

Upgrade torch version to 2.5 in dockerfile for quantization CI (#34924)

Browse Source 
* Upgrade Torch 2.5

* uncomment
This commit is contained in:
Mohamed Mekkouri 2024-11-25 17:38:20 +01:00 committed by GitHub
parent a830df2909
commit b76a292bde
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docker/transformers-quantization-latest-gpu/Dockerfile
View File

@ -9,7 +9,7 @@ SHELL ["sh", "-lc"]
# The following `ARG` are mainly used to specify the versions explicitly & directly in this docker file, and not meant # 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). # to be used as arguments for docker build (so far).
ARG PYTORCH='2.4.1' ARG PYTORCH='2.5.1'
# Example: `cu102`, `cu113`, etc. # Example: `cu102`, `cu113`, etc.
ARG CUDA='cu118' ARG CUDA='cu118'