From 988ec52a509b0c38a7a5077c0ac01432a3ca13d7 Mon Sep 17 00:00:00 2001
From: Xuefen <37746681+Sprinter1999@users.noreply.github.com>
Date: Sun, 19 May 2024 16:14:56 +0800
Subject: [PATCH 1/2] Create gdown_download_checkpoints.sh

---
 scripts/gdown_download_checkpoints.sh | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 scripts/gdown_download_checkpoints.sh

diff --git a/scripts/gdown_download_checkpoints.sh b/scripts/gdown_download_checkpoints.sh
new file mode 100644
index 0000000..ee20f35
--- /dev/null
+++ b/scripts/gdown_download_checkpoints.sh
@@ -0,0 +1,21 @@
+#!/bin/bash
+
+SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
+
+mkdir -p $SCRIPT_DIR/../checkpoints
+
+pip install gdown
+pip install --upgrade gdown
+
+wget https://huggingface.co/stabilityai/stable-diffusion-2-1-base/resolve/main/v2-1_512-ema-pruned.ckpt $SCRIPT_DIR/../checkpoints/v2-1_512-ema-pruned.ckpt
+
+gdown https://storage.googleapis.com/sfr-hive-data-research/checkpoints/hive_rw_condition.ckpt -O $SCRIPT_DIR/../checkpoints/hive_rw_condition.ckpt
+gdown https://storage.googleapis.com/sfr-hive-data-research/checkpoints/hive_v2_rw_condition.ckpt -O $SCRIPT_DIR/../checkpoints/hive_v2_rw_condition.ckpt
+gdown https://storage.googleapis.com/sfr-hive-data-research/checkpoints/hive_rw.ckpt -O $SCRIPT_DIR/../checkpoints/hive_rw.ckpt
+gdown https://storage.googleapis.com/sfr-hive-data-research/checkpoints/hive_v2_rw.ckpt -O $SCRIPT_DIR/../checkpoints/hive_v2_rw.ckpt
+
+
+#gcloud storage cp gs://sfr-hive-data-research/checkpoints/hive_rw_condition.ckpt $SCRIPT_DIR/../checkpoints/hive_rw_condition.ckpt
+#gcloud storage cp gs://sfr-hive-data-research/checkpoints/hive_v2_rw_condition.ckpt $SCRIPT_DIR/../checkpoints/hive_v2_rw_condition.ckpt
+#gcloud storage cp gs://sfr-hive-data-research/checkpoints/hive_rw.ckpt $SCRIPT_DIR/../checkpoints/hive_rw.ckpt
+#gcloud storage cp gs://sfr-hive-data-research/checkpoints/hive_v2_rw.ckpt $SCRIPT_DIR/../checkpoints/hive_v2_rw.ckpt

From be1068e91a11eb3b9234e8b6536a3eed91218ccc Mon Sep 17 00:00:00 2001
From: Xuefen <37746681+Sprinter1999@users.noreply.github.com>
Date: Sun, 19 May 2024 16:18:11 +0800
Subject: [PATCH 2/2] Update gdown_download_checkpoints.sh

---
 scripts/gdown_download_checkpoints.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/gdown_download_checkpoints.sh b/scripts/gdown_download_checkpoints.sh
index ee20f35..b677a83 100644
--- a/scripts/gdown_download_checkpoints.sh
+++ b/scripts/gdown_download_checkpoints.sh
@@ -4,6 +4,8 @@ SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
 
 mkdir -p $SCRIPT_DIR/../checkpoints
 
+#TODO: we can download the checkpoints with gdown tool, sometimes it is hard to use the gcloud tool
+
 pip install gdown
 pip install --upgrade gdown