Update app.py

app.py

@@ -78,7 +78,7 @@ def parse_args():
     parser.add_argument(
         "--pretrained_model_name_or_path", 
         type=str,
-        default='botp/stable-diffusion-v1-5', # no need to modify this  
+        default='stable-diffusion-v1-5/stable-diffusion-v1-5', # no need to modify this  
         help="Path to pretrained model or model identifier from huggingface.co/models. Please do not modify this.",
     )
     parser.add_argument(
@@ -361,20 +361,20 @@ if accelerator.is_main_process:
 
 # Load scheduler, tokenizer and models.
 tokenizer15 = CLIPTokenizer.from_pretrained(
-    'botp/stable-diffusion-v1-5', subfolder="tokenizer", revision=args.revision
+    'stable-diffusion-v1-5/stable-diffusion-v1-5', subfolder="tokenizer", revision=args.revision
 )
 tokenizer21 = CLIPTokenizer.from_pretrained(
     'stabilityai/stable-diffusion-2-1', subfolder="tokenizer", revision=args.revision
 )
 
 text_encoder15 = CLIPTextModel.from_pretrained(
-    'botp/stable-diffusion-v1-5', subfolder="text_encoder", revision=args.revision
+    'stable-diffusion-v1-5/stable-diffusion-v1-5', subfolder="text_encoder", revision=args.revision
 )
 text_encoder21 = CLIPTextModel.from_pretrained(
     'stabilityai/stable-diffusion-2-1', subfolder="text_encoder", revision=args.revision
 )
 
-vae15 = AutoencoderKL.from_pretrained('botp/stable-diffusion-v1-5', subfolder="vae", revision=args.revision).cuda()
+vae15 = AutoencoderKL.from_pretrained('stable-diffusion-v1-5/stable-diffusion-v1-5', subfolder="vae", revision=args.revision).cuda()
 unet15 = UNet2DConditionModel.from_pretrained(
     './textdiffuser-ckpt/diffusion_backbone_1.5', subfolder="unet", revision=None 
 ).cuda() 
@@ -384,7 +384,7 @@ unet21 = UNet2DConditionModel.from_pretrained(
     './textdiffuser-ckpt/diffusion_backbone_2.1', subfolder="unet", revision=None 
 ).cuda() 
 
-scheduler15 = DDPMScheduler.from_pretrained('botp/stable-diffusion-v1-5', subfolder="scheduler") 
+scheduler15 = DDPMScheduler.from_pretrained('stable-diffusion-v1-5/stable-diffusion-v1-5', subfolder="scheduler") 
 scheduler21 = DDPMScheduler.from_pretrained('stabilityai/stable-diffusion-2-1', subfolder="scheduler")