diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml index 623083db2..f3ad69570 100644 --- a/.github/workflows/base.yml +++ b/.github/workflows/base.yml @@ -26,7 +26,7 @@ jobs: pytorch: 2.0.0 axolotl_extras: - cuda: "117" - cuda_version: 11.7.0 + cuda_version: 11.7.1 python_version: "3.9" pytorch: 1.13.1 axolotl_extras: diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 033199154..07f25cac6 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -30,7 +30,7 @@ jobs: pytorch: 2.0.0 axolotl_extras: gptq - cuda: cu117 - cuda_version: 11.7.0 + cuda_version: 11.7.1 python_version: "3.9" pytorch: 1.13.1 axolotl_extras: @@ -85,7 +85,7 @@ jobs: pytorch: 2.0.0 axolotl_extras: gptq - cuda: cu117 - cuda_version: 11.7.0 + cuda_version: 11.7.1 python_version: "3.9" pytorch: 1.13.1 axolotl_extras: