diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index d2a3e792..d8b3f288 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -31,7 +31,7 @@ jobs:
       group: ${{ github.workflow }}-build-and-push-image-${{ inputs.hardware }}-${{ github.head_ref || github.run_id }}
       cancel-in-progress: true
     runs-on:
-      group: aws-highmemory-32-plus-priv
+      group: aws-highmemory-64-plus-priv
     permissions:
       contents: write
       packages: write
diff --git a/.github/workflows/build_trtllm.yaml b/.github/workflows/build_trtllm.yaml
index a66f3cfc..3db358e4 100644
--- a/.github/workflows/build_trtllm.yaml
+++ b/.github/workflows/build_trtllm.yaml
@@ -1,12 +1,6 @@
 name: Build TensorRT-LLM
 on:
   workflow_call:
-    inputs:
-      runs-on:
-        type: string
-        description: "Which instance type to use to run the workflow"
-        required: true
-
     outputs:
       docker_image:
         description: "Reference to the Docker Image build by this workflow"