upgrade pytorch to 2.5.1 (#2024)

This commit is contained in:
Wing Lian
2024-11-08 10:46:24 -05:00
committed by GitHub
parent 035e9f9dd7
commit 3cb2d75de1
5 changed files with 9 additions and 9 deletions

View File

@@ -31,7 +31,7 @@ jobs:
- cuda: 124
cuda_version: 12.4.1
python_version: "3.11"
pytorch: 2.5.0
pytorch: 2.5.1
axolotl_extras:
num_gpus: 2
nightly_build: "true"