use pytorch 2.0.1

This commit is contained in:
Wing Lian
2023-07-20 00:47:13 -04:00
parent 28fd429bcf
commit c58034d48c
2 changed files with 9 additions and 9 deletions

View File

@@ -18,12 +18,12 @@ jobs:
- cuda: "118"
cuda_version: 11.8.0
python_version: "3.9"
pytorch: 2.0.0
pytorch: 2.0.1
axolotl_extras:
- cuda: "118"
cuda_version: 11.8.0
python_version: "3.10"
pytorch: 2.0.0
pytorch: 2.0.1
axolotl_extras:
- cuda: "117"
cuda_version: 11.7.1
@@ -33,7 +33,7 @@ jobs:
- cuda: "118"
cuda_version: 11.8.0
python_version: "3.9"
pytorch: 2.0.0
pytorch: 2.0.1
axolotl_extras: gptq
steps:
- name: Checkout

View File

@@ -17,17 +17,17 @@ jobs:
- cuda: cu118
cuda_version: 11.8.0
python_version: "3.9"
pytorch: 2.0.0
pytorch: 2.0.1
axolotl_extras:
- cuda: cu118
cuda_version: 11.8.0
python_version: "3.10"
pytorch: 2.0.0
pytorch: 2.0.1
axolotl_extras:
- cuda: cu118
cuda_version: 11.8.0
python_version: "3.9"
pytorch: 2.0.0
pytorch: 2.0.1
axolotl_extras: gptq
- cuda: cu117
cuda_version: 11.7.1
@@ -72,17 +72,17 @@ jobs:
- cuda: cu118
cuda_version: 11.8.0
python_version: "3.9"
pytorch: 2.0.0
pytorch: 2.0.1
axolotl_extras:
- cuda: cu118
cuda_version: 11.8.0
python_version: "3.10"
pytorch: 2.0.0
pytorch: 2.0.1
axolotl_extras:
- cuda: cu118
cuda_version: 11.8.0
python_version: "3.9"
pytorch: 2.0.0
pytorch: 2.0.1
axolotl_extras: gptq
- cuda: cu117
cuda_version: 11.7.1