Update app.py

app.py

@@ -37,7 +37,7 @@ vae = AutoencoderKL.from_pretrained(
 )
 
 # Ensure model and scheduler are initialized in GPU-enabled function
-pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16).to(device)
+pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae, torch_dtype="auto").to(device)
 
 pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)