fix runner size

Steboss · Steboss · commit d3500bdf9c59 · 2025-02-18T16:48:14.000Z
diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml
@@ -187,6 +187,7 @@ jobs:
       BASE_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAG_MEALKIT }}
       CONTAINER_NAME: axlearn
       DOCKERFILE: .github/container/Dockerfile.axlearn
+      RUNNER_SIZE: large
     secrets: inherit
 
   collect-docker-tags:
