From 4ae32cf07701185c77753f88980e8804d7c14e85 Mon Sep 17 00:00:00 2001
From: sumof2primes <sumof2primes@gmail.com>
Date: Mon, 13 Mar 2023 00:24:36 +0900
Subject: [PATCH] =?UTF-8?q?=EA=B5=AC=EA=B8=80=20=EB=93=9C=EB=9D=BC?=
 =?UTF-8?q?=EC=9D=B4=EB=B8=8C=20=EB=A7=88=EC=9A=B4=ED=8C=85=20=EC=98=B5?=
 =?UTF-8?q?=EC=85=98=EC=9D=84=20=EC=A0=9C=EA=B3=B5=ED=95=B4=20=EC=A3=BC?=
 =?UTF-8?q?=ED=94=BC=ED=84=B0=20=EB=A1=9C=EC=BB=AC=20=EB=9F=B0=ED=83=80?=
 =?UTF-8?q?=EC=9E=84=EC=97=90=EC=84=9C=EB=8F=84=20=EC=8B=A4=ED=96=89=20?=
 =?UTF-8?q?=EA=B0=80=EB=8A=A5=ED=95=98=EB=8F=84=EB=A1=9D=20=EC=88=98?=
 =?UTF-8?q?=EC=A0=95=20=20-=20=EC=98=88)=20RunPod=20=EB=A1=9C=EC=BB=AC=20?=
 =?UTF-8?q?=EB=9F=B0=ED=83=80=EC=9E=84=EC=97=90=20=EC=97=B0=EA=B2=B0?=
 =?UTF-8?q?=ED=95=B4=20=EC=BD=94=EB=9E=A9=20=EB=85=B8=ED=8A=B8=EB=B6=81=20?=
 =?UTF-8?q?=EC=82=AC=EC=9A=A9=20=EA=B0=80=EB=8A=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 easy-stable-diffusion-downloader.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/easy-stable-diffusion-downloader.py b/easy-stable-diffusion-downloader.py
index e824930..0809cc4 100644
--- a/easy-stable-diffusion-downloader.py
+++ b/easy-stable-diffusion-downloader.py
@@ -8,7 +8,8 @@
 from typing import Union, List, Dict
 from IPython.display import display
 from ipywidgets import widgets
-from google.colab import drive, runtime
+
+OPTIONS = {}
 
 # fmt: off
 #@title
@@ -20,6 +21,11 @@
 #@markdown ##### <font color="orange">***다운로드가 끝나면 자동으로 코랩 런타임을 종료할지?***</font>
 DISCONNECT_RUNTIME = True  #@param {type:"boolean"}
 
+#@markdown ##### <font color="orange">***구글 드라이브와 동기화할지?***</font>
+#@markdown <font color="red">**주의**</font>: 동기화 전 남은 용량이 충분한지 확인 필수 (5GB 이상)
+USE_GOOGLE_DRIVE = True  #@param {type:"boolean"}
+OPTIONS['USE_GOOGLE_DRIVE'] = USE_GOOGLE_DRIVE
+
 # fmt: on
 
 # 인터페이스 요소
@@ -42,14 +48,16 @@
 
 
 # 파일 경로
-workspace_dir = Path('drive', 'MyDrive', WORKSPACE)
+workspace_dir = Path('drive', 'MyDrive', WORKSPACE) if OPTIONS['USE_GOOGLE_DRIVE'] else Path(WORKSPACE)
 sd_model_dir = workspace_dir.joinpath('models', 'Stable-diffusion')
 sd_embedding_dir = workspace_dir.joinpath('embeddings')
 vae_dir = workspace_dir.joinpath('models', 'VAE')
 
 # 구글 드라이브 마운팅
-with output:
-    drive.mount('drive')
+if OPTIONS['USE_GOOGLE_DRIVE']:
+    from google.colab import drive, runtime
+    with output:
+        drive.mount('drive')
 
 sd_model_dir.mkdir(0o777, True, True)
 sd_embedding_dir.mkdir(0o777, True, True)