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' \