Fix workflow

Author: aybchan

.github/workflows/jax-vllm-offloading/gke_transfer.yml renamed to .github/workflows/jax-vllm-offloading-gke_transfer.yml

@@ -17,16 +17,16 @@ on:
     paths:
       - 'jax-inference-offloading/**'
       - '.github/gke-workflow/jax-vllm-offloading/**'
-      - '.github/workflows/jax-vllm-offloading/**'
-      - '.github/workflows/jax-vllm-offloading.yml'
+      - '.github/workflows/jax-vllm-offloading*.yml'
 
 jobs:
   gke:
+    runs-on: gke-a3mega
+
     steps:
-      jax-vllm-offloading-transfer-gke-xpk:
-      - uses: .github/workflows/jax-vllm-offloading/gke_transfer.yml
+      - name: jax-vllm-offloading-transfer-gke-xpk
+        uses: .github/workflows/jax-vllm-offloading-gke_transfer.yml
         # needs: build # .github/workflows/jio.yaml
         with:
         # JAX_VLLM_OFFLOADING_IMAGE: ${{ needs.build.outputs.DOCKER_TAG_FINAL }}
           JAX_VLLM_OFFLOADING_IMAGE: ghcr.io/nvidia/jax-toolbox-internal:19461214142-jio-amd64
-        secrets: inherit