Remove device assign

app.py

@@ -52,7 +52,7 @@ is_attention_slicing_enabled = True
 
 # Load model
 dtype = torch.float16
-device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+# device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
 
 model_path = "stabilityai/stable-diffusion-xl-base-1.0"