Update app.py

app.py

@@ -1,6 +1,7 @@
 from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, DPMSolverMultistepScheduler
 import gradio as gr
 import torch
+pipe = pipe.to("cuda")
 from PIL import Image
 
 model_id = 'hassanblend/HassanBlend1.5'
@@ -19,10 +20,12 @@ scheduler = DPMSolverMultistepScheduler(
     lower_order_final=True,
 )
 
-pipe = StableDiffusionPipeline.from_pretrained(
-  model_id,
-  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-  scheduler=scheduler)
+vae = diffusers.AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
+pipe = diffusers.StableDiffusionPipeline.from_pretrained(
+    "hassanblend/HassanBlend1.5.1.2",
+    torch_dtype=torch.float16,
+    vae=vae,
+)
 
 pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
   model_id,