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: "118"
cuda_version: 11.8.0 cuda_version: 11.8.0
python_version: "3.9" python_version: "3.9"
pytorch: 2.0.0 pytorch: 2.0.1
axolotl_extras: axolotl_extras:
- cuda: "118" - cuda: "118"
cuda_version: 11.8.0 cuda_version: 11.8.0
python_version: "3.10" python_version: "3.10"
pytorch: 2.0.0 pytorch: 2.0.1
axolotl_extras: axolotl_extras:
- cuda: "117" - cuda: "117"
cuda_version: 11.7.1 cuda_version: 11.7.1
@@ -33,7 +33,7 @@ jobs:
- cuda: "118" - cuda: "118"
cuda_version: 11.8.0 cuda_version: 11.8.0
python_version: "3.9" python_version: "3.9"
pytorch: 2.0.0 pytorch: 2.0.1
axolotl_extras: gptq axolotl_extras: gptq
steps: steps:
- name: Checkout - name: Checkout

View File

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