test

huggingface · Mar 9, 2025 · c73e5f4 · c73e5f4
1 parent cc9b078
commit c73e5f4
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/.github/workflows/test_upstream.yml → .github/workflows/upstream.yml b/.github/workflows/test_upstream.yml → .github/workflows/upstream.yml
@@ -14,8 +14,7 @@ concurrency:
 jobs:
   transformers:
     name: Upstream Transformers
-    runs-on:
-      group: aws-dl1-24xlarge
+    runs-on: [self-hosted, linux, x64, gaudi2, fast]
 
     container:
       image: docker://vault.habana.ai/gaudi-docker/1.20.0/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest
@@ -85,8 +84,7 @@ jobs:
 
   accelerate:
     name: Upstream Accelerate
-    runs-on:
-      group: aws-dl1-24xlarge
+    runs-on: [self-hosted, linux, x64, gaudi2, fast]
 
     container:
       image: docker://vault.habana.ai/gaudi-docker/1.20.0/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest