revert previous change and build ax images w docker on gpu (#371)

2023-08-12 20:23:00 -04:00
parent c3fde36ada
commit 918f1b0dfb
1 changed files with 2 additions and 2 deletions
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -29,7 +29,7 @@ jobs:
            python_version: "3.9"
            pytorch: 2.0.1
            axolotl_extras: gptq
-    runs-on: ubuntu-latest
+    runs-on: self-hosted
    steps:
      - name: Checkout
        uses: actions/checkout@v3
@@ -77,7 +77,7 @@ jobs:
            python_version: "3.9"
            pytorch: 2.0.1
            axolotl_extras: gptq
-    runs-on: ubuntu-latest
+    runs-on: self-hosted
    steps:
      - name: Checkout
        uses: actions/checkout@v3