Fixed error

app.py

@@ -42,7 +42,7 @@ def inference(
     device = torch.device("cuda")
     model_path = "stabilityai/stable-diffusion-xl-base-1.0"
     scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
-    model = StableDiffusionXLPipeline.from_pretrained(model_path, scheduler=scheduler, device=device, torch_dtype=torch.float16)
+    model = StableDiffusionXLPipeline.from_pretrained(model_path, scheduler=scheduler, torch_dtype=torch.float16).to(device)
     model.unet.set_default_attn_processor()
     model.enable_sequential_cpu_offload()