Update runner version in GitHub Actions workflow

.github/workflows/pull.yml

@@ -638,7 +638,7 @@ jobs:
   test-mps:
     strategy:
       matrix:
-        runner: [macos-m1-stable ]
+        runner: [macos-14-xlarge]
     runs-on: ${{matrix.runner}}
     steps:
       - name: Checkout repo
@@ -695,6 +695,16 @@ jobs:
           export MODEL_PATH=checkpoints/stories15M/stories15M.pt
           export MODEL_NAME=stories15M
           export MODEL_DIR=/tmp
+          python - <<'PY'
+          import torch
+          if torch.backends.mps.is_available():
+              try:
+                  print("MPS current:", torch.mps.current_allocated_memory() / (1024**3), "GiB")
+                  print("MPS driver :", torch.mps.driver_allocated_memory()  / (1024**3), "GiB")
+                  print("MPS peak   :", torch.mps.driver_allocated_memory()  / (1024**3), "GiB")
+              except Exception:
+                  pass
+          PY
 
           python3 torchchat.py generate --device mps --checkpoint-path ${MODEL_PATH} --temperature 0
 
@@ -768,7 +778,7 @@ jobs:
   test-mps-dtype:
     strategy:
       matrix:
-        runner: [macos-m1-stable ]
+        runner: [macos-14-xlarge]
     runs-on: ${{matrix.runner}}
     steps:
       - name: Checkout repo
@@ -1213,7 +1223,7 @@ jobs:
   test-torchao-experimental-mps:
     strategy:
       matrix:
-        runner: [macos-m1-stable]
+        runner: [macos-14-xlarge]
     runs-on: ${{matrix.runner}}
     steps:
       - name: Checkout repo