Update app.py

app.py

@@ -88,6 +88,10 @@ 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()
+
 to_cuda(torch, pipe)
 
 def to_cuda(torch, pipe):
@@ -240,7 +244,7 @@ def img_to_img(model_path, prompt, n_images, neg_prompt, img, strength, guidance
               )
           # pipe = pipe.to("cpu")
           # pipe = current_model.pipe_i2i
-        
+
 to_cuda(torch, pipe)
 
 def to_cuda(torch, pipe):