From 918f1b0dfb395cbd2e5785d18d0a1cca5e82f549 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sat, 12 Aug 2023 20:23:00 -0400 Subject: [PATCH] revert previous change and build ax images w docker on gpu (#371) --- .github/workflows/main.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index b5c577042..9446e03a1 100644 --- 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