Update app.py

app.py

@@ -37,11 +37,11 @@ if torch.cuda.is_available():
         use_safetensors=True,
         torch_dtype=torch.float32
     )
-    # vae = AutoencoderKL.from_pretrained(vae_model_id, torch_dtype=torch.float16)
+    vae = AutoencoderKL.from_pretrained(vae_model_id, torch_dtype=torch.float16)
     pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
         base_model_id, 
         controlnet=controlnet, 
-        # vae=vae,
+        vae=vae,
         # variant="fp16",
         use_safetensors=True,
         torch_dtype=torch.float32,