updte gke yaml

Signed-off-by: Juncheng Gu <jcgu@google.com>

Author: juncgu-google

examples/gke/benchmarks/deploy-cpu-offload.yaml

@@ -30,8 +30,10 @@ spec:
               key: token
         - name: SKIP_JAX_PRECOMPILE
           value: "1"
-        - name: TPU_OFFLOAD_CPU_CACHE_SIZE_GB
-          value: "1024"
+        - name: TPU_OFFLOAD_NUM_CPU_CHUNKS
+          value: "4096"
+        - name: TPU_OFFLOAD_NUM_STAGING_BLOCKS
+          value: "256"
         ports:
         - containerPort: 8000
         resources: