diff --git a/configs/sd2_gradio.yaml b/configs/sd2_gradio.yaml
index f90b4ef..064971a 100644
--- a/configs/sd2_gradio.yaml
+++ b/configs/sd2_gradio.yaml
@@ -1,5 +1,5 @@
 base_model_type: sd2
-base_model_path: stabilityai/stable-diffusion-2-1-base
+base_model_path: sd2-community/stable-diffusion-2-1-base
 model_t: 200
 coeff_t: 200
 lora_rank: 256
diff --git a/configs/sd2_train.yaml b/configs/sd2_train.yaml
index 449d187..823aebf 100644
--- a/configs/sd2_train.yaml
+++ b/configs/sd2_train.yaml
@@ -70,7 +70,7 @@ data_config:
         resize_back: true
 
 base_model_type: sd2
-base_model_path: stabilityai/stable-diffusion-2-1-base
+base_model_path: sd2-community/stable-diffusion-2-1-base
 model_t: 200
 coeff_t: 200
 lora_rank: 256