tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 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: "121"
cuda_version: 12.1.0 cuda_version: 12.1.0
python_version: "3.11" 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" torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
- cuda: "121" - cuda: "121"
cuda_version: 12.1.0 cuda_version: 12.1.0

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

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

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

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

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

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

2
README.md
View File

@@ -302,7 +302,7 @@ Write a job description in YAML as below:
# dstack.yaml # dstack.yaml
type: task type: task
image: winglian/axolotl-cloud:main-20240429-py3.11-cu121-2.2.1 image: winglian/axolotl-cloud:main-20240429-py3.11-cu121-2.2.2
env: env:
- HUGGING_FACE_HUB_TOKEN - HUGGING_FACE_HUB_TOKEN