Update app.py

app.py

@@ -87,9 +87,9 @@ else:
       scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler")
       )
 
-        if torch.cuda.is_available():
-          pipe = pipe.to("cuda")
-          pipe.enable_xformers_memory_efficient_attention()
+    if torch.cuda.is_available():
+        pipe = pipe.to("cuda")
+        pipe.enable_xformers_memory_efficient_attention()
 
 device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"