tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

update torch 2.2.1 -> 2.2.2 (#1622)

Browse Source
This commit is contained in:
Wing Lian
2024-05-15 09:45:27 -04:00
committed by GitHub
parent 81da7d2531
commit 3319780300
5 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/base.yml vendored
View File

@@ -30,7 +30,7 @@ jobs:
- cuda: "121"
cuda_version: 12.1.0
python_version: "3.11"
pytorch: 2.2.1
pytorch: 2.2.2
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
- cuda: "121"
cuda_version: 12.1.0

4
.github/workflows/main.yml vendored
View File

@@ -28,7 +28,7 @@ jobs:
- cuda: 121
cuda_version: 12.1.0
python_version: "3.11"
pytorch: 2.2.1
pytorch: 2.2.2
axolotl_extras:
- cuda: 121
cuda_version: 12.1.0
@@ -89,7 +89,7 @@ jobs:
- cuda: 121
cuda_version: 12.1.0
python_version: "3.11"
pytorch: 2.2.1
pytorch: 2.2.2
axolotl_extras:
- cuda: 121
cuda_version: 12.1.0

4
.github/workflows/nightlies.yml vendored
View File

@@ -27,7 +27,7 @@ jobs:
- cuda: 121
cuda_version: 12.1.0
python_version: "3.11"
pytorch: 2.2.1
pytorch: 2.2.2
axolotl_extras:
- cuda: 121
cuda_version: 12.1.0
@@ -89,7 +89,7 @@ jobs:
- cuda: 121
cuda_version: 12.1.0
python_version: "3.11"
pytorch: 2.2.1
pytorch: 2.2.2
axolotl_extras:
- cuda: 121
cuda_version: 12.1.0

2
.github/workflows/tests.yml vendored
View File

@@ -82,7 +82,7 @@ jobs:
- cuda: 121
cuda_version: 12.1.0
python_version: "3.11"
pytorch: 2.2.1
pytorch: 2.2.2
num_gpus: 1
steps:
- name: Checkout