Update app.py

app.py

@@ -12,7 +12,7 @@ import torch
 import numpy as np
 import cv2
 
-#vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 controlnet = ControlNetModel.from_pretrained(
     "diffusers/controlnet-canny-sdxl-1.0",
     torch_dtype=torch.float16
@@ -21,7 +21,7 @@ controlnet = ControlNetModel.from_pretrained(
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     controlnet=controlnet,
-    #vae=vae,
+    vae=vae,
     torch_dtype=torch.float16, 
     variant="fp16",
     use_safetensors=True