Update app.py

app.py

@@ -35,7 +35,7 @@ text_encoder = CLIPTextModel.from_pretrained(
     'JingyeChen22/textdiffuser2-full-ft-inpainting', subfolder="text_encoder"
 ).cuda().half()
 tokenizer = CLIPTokenizer.from_pretrained(
-    'runwayml/stable-diffusion-v1-5', subfolder="tokenizer"
+    'botp/stable-diffusion-v1-5', subfolder="tokenizer"
 )
 
 #### additional tokens are introduced, including coordinate tokens and character tokens
@@ -51,7 +51,7 @@ for c in alphabet:
 print(len(tokenizer))
 print('***************')
 
-vae = AutoencoderKL.from_pretrained('runwayml/stable-diffusion-v1-5', subfolder="vae").half().cuda()
+vae = AutoencoderKL.from_pretrained('botp/stable-diffusion-v1-5', subfolder="vae").half().cuda()
 unet = UNet2DConditionModel.from_pretrained(
     'JingyeChen22/textdiffuser2-full-ft-inpainting', subfolder="unet"
 ).half().cuda()
@@ -515,7 +515,7 @@ def text_to_image(guest_id, i, orig_i, prompt,keywords,positive_prompt,radio,sli
         prompts_cond = torch.Tensor(prompts_cond).long().cuda()
         prompts_nocond = torch.Tensor(prompts_nocond).long().cuda()
 
-        scheduler = DDPMScheduler.from_pretrained('runwayml/stable-diffusion-v1-5', subfolder="scheduler") 
+        scheduler = DDPMScheduler.from_pretrained('botp/stable-diffusion-v1-5', subfolder="scheduler") 
         scheduler.set_timesteps(slider_step) 
         noise = torch.randn((slider_batch, 4, 64, 64)).to("cuda").half()
         input = noise