diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 677e722a..f81b1b63 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -133,7 +133,7 @@ jobs:
   build-and-push-sagemaker-image:
     needs:
       - build-and-push-image
-    runs-on: ubuntu-latest
+    runs-on: large
     steps:
       - name: Checkout repository
         uses: actions/checkout@v3