runner

AI-Hypercomputer · Jul 25, 2024 · ef6dfe0 · ef6dfe0
1 parent e048add
commit ef6dfe0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/UploadDockerImages.yml b/.github/workflows/UploadDockerImages.yml
@@ -23,7 +23,7 @@ on:
 
 jobs:
   build-image:
-    runs-on: ["self-hosted", "e2"]
+    runs-on: ["self-hosted", "e2", "cpu"]
     steps:
     - uses: actions/checkout@v3
     - name: build jax stable stack image

diff --git a/requirements_with_jax_stable_stack.txt b/requirements_with_jax_stable_stack.txt
@@ -1,5 +1,5 @@
 # Requirements for Building the MaxDifussion Docker Image
-# These requirements are additional to the dependencies present in the JAX SS base image.
+# These requirements are additional to the dependencies present in the JAX Stable Stack base image.
 absl-py
 transformers>=4.25.1
 datasets