From c17bbd00b6a8dcb35ecfc99b926f44c20f55b0a9 Mon Sep 17 00:00:00 2001
From: gurusai-voleti <gvoleti@google.com>
Date: Tue, 27 Oct 2026 04:29:53 +0000
Subject: [PATCH] chore: migrate gsutil usage to gcloud storage

---
 benchmarks/vbench/run_gpu_eval.sh                             | 3 ---
 src/maxdiffusion/pedagogical_examples/save_sd_checkpoint.py   | 2 +-
 src/maxdiffusion/pedagogical_examples/save_sdxl_checkpoint.py | 2 +-
 3 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/benchmarks/vbench/run_gpu_eval.sh b/benchmarks/vbench/run_gpu_eval.sh
index c1d7b48bb..6a3f5df75 100644
--- a/benchmarks/vbench/run_gpu_eval.sh
+++ b/benchmarks/vbench/run_gpu_eval.sh
@@ -397,9 +397,6 @@ gcs_cp() {
   if have gcloud && gcloud storage cp "$@"; then
     return 0
   fi
-  if have gsutil && gsutil -m cp "$@"; then
-    return 0
-  fi
   return 1
 }
 
diff --git a/src/maxdiffusion/pedagogical_examples/save_sd_checkpoint.py b/src/maxdiffusion/pedagogical_examples/save_sd_checkpoint.py
index a54f19349..f9b5fd2be 100644
--- a/src/maxdiffusion/pedagogical_examples/save_sd_checkpoint.py
+++ b/src/maxdiffusion/pedagogical_examples/save_sd_checkpoint.py
@@ -16,7 +16,7 @@
 
 """Load and save a checkpoint. This is useful for uploading checkpoints to gcs
 and later loading them from gcs directly.
-After calling this script, use gsutil to upload the weights to a bucket:
+After calling this script, use gcloud storage to upload the weights to a bucket:
 gcloud storage cp --recursive sd-model-finetuned gs://<your-bucket>/sd_checkpoint/
 """
 
diff --git a/src/maxdiffusion/pedagogical_examples/save_sdxl_checkpoint.py b/src/maxdiffusion/pedagogical_examples/save_sdxl_checkpoint.py
index 46caa7450..21de0d743 100644
--- a/src/maxdiffusion/pedagogical_examples/save_sdxl_checkpoint.py
+++ b/src/maxdiffusion/pedagogical_examples/save_sdxl_checkpoint.py
@@ -16,7 +16,7 @@
 
 """Load and save a checkpoint. This is useful for uploading checkpoints to gcs
 and later loading them from gcs directly.
-After calling this script, use gsutil to upload the weights to a bucket:
+After calling this script, use gcloud storage to upload the weights to a bucket:
 gcloud storage cp --recursive sdxl-model-finetuned gs://<your-bucket>/sdxl_1.0_base/
 """