use RUNNER to set ip

ModelCloud · Dec 4, 2024 · 0100a13 · 0100a13
1 parent 28411cb
commit 0100a13
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
@@ -9,6 +9,10 @@ on:
   repository_dispatch:
   workflow_dispatch:
 
+env:
+  CUDA_DEVICE_ORDER: PCI_BUS_ID
+  RUNNER: 10.0.14.248
+
 jobs:
   build:
     runs-on: ${{ matrix.os }}
@@ -69,7 +73,7 @@ jobs:
   gpu:
     runs-on: self-hosted
     container:
-      image: 10.0.14.248:5000/modelcloud/gptqmodel:compiler_cuda124-torch2.5.1-python311
+      image: ${RUNNER}:5000/modelcloud/gptqmodel:compiler_cuda124-torch2.5.1-python311
     steps:
       - uses: actions/checkout@v4