From 34ea2b474a4c1fde414bc96bad860ff7f3fb2e7e Mon Sep 17 00:00:00 2001
From: Wang Qiang <37444407+XDUWQ@users.noreply.github.com>
Date: Thu, 27 Jul 2023 20:40:38 +0800
Subject: [PATCH] Upgrade stable diffusion version to a more powerful version
 2.1 (#415)

---
 .../pytorch/stable_diffusion/dreambooth/run_train_dreambooth.sh | 2 +-
 examples/pytorch/stable_diffusion/lora/run_train_lora.sh        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/pytorch/stable_diffusion/dreambooth/run_train_dreambooth.sh b/examples/pytorch/stable_diffusion/dreambooth/run_train_dreambooth.sh
index dcb3b2e6..03c51ffb 100644
--- a/examples/pytorch/stable_diffusion/dreambooth/run_train_dreambooth.sh
+++ b/examples/pytorch/stable_diffusion/dreambooth/run_train_dreambooth.sh
@@ -1,5 +1,5 @@
 PYTHONPATH=. torchrun examples/pytorch/stable_diffusion/dreambooth/finetune_stable_diffusion_dreambooth.py \
-    --model 'AI-ModelScope/stable-diffusion-v1-5' \
+    --model 'AI-ModelScope/stable-diffusion-v2-1' \
     --model_revision 'v1.0.8' \
     --work_dir './tmp/dreambooth_diffusion' \
     --train_dataset_name 'buptwq/lora-stable-diffusion-finetune' \
diff --git a/examples/pytorch/stable_diffusion/lora/run_train_lora.sh b/examples/pytorch/stable_diffusion/lora/run_train_lora.sh
index bf62f833..d4f1b07d 100644
--- a/examples/pytorch/stable_diffusion/lora/run_train_lora.sh
+++ b/examples/pytorch/stable_diffusion/lora/run_train_lora.sh
@@ -1,5 +1,5 @@
 PYTHONPATH=. torchrun examples/pytorch/stable_diffusion/lora/finetune_stable_diffusion_lora.py \
-    --model 'AI-ModelScope/stable-diffusion-v1-5' \
+    --model 'AI-ModelScope/stable-diffusion-v2-1' \
     --model_revision 'v1.0.9' \
     --prompt "a dog" \
     --work_dir './tmp/lora_diffusion' \