Update app.py

app.py

@@ -35,16 +35,16 @@ if torch.cuda.is_available():
         controlnet_model_id, 
         variant="fp16",
         use_safetensors=True,
-        torch_dtype=torch.float32
+        torch_dtype=torch.bfloat16
     )
-    vae = AutoencoderKL.from_pretrained(vae_model_id, torch_dtype=torch.float32)
+    vae = AutoencoderKL.from_pretrained(vae_model_id, torch_dtype=torch.bfloat16)
     pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
         base_model_id, 
         controlnet=controlnet, 
         vae=vae,
         variant="fp16",
         use_safetensors=True,
-        torch_dtype=torch.float32,
+        torch_dtype=torch.bfloat16,
     )
     pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
     pipe.to(device)