Update app.py

app.py

@@ -15,6 +15,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
   model_id,
   torch_dtype=torch.float32 if torch.cuda.is_available() else torch.float32,
   scheduler=scheduler)
+pipe.enable_xformers_memory_efficient_attention()
 
 pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
   model_id,
@@ -43,15 +44,15 @@ def inference(prompt, guidance, steps, image_size="Square", seed=0, img=None, st
   if(image_size=="Portrait"):
       height=1024
       width=768
-      pipe.enable_attention_slicing()
+      #pipe.enable_attention_slicing()
   elif(image_size=="Landscape"):
       height=768
       width=1024  
-      pipe.enable_attention_slicing()
+      #pipe.enable_attention_slicing()
   else:
       height=768
       width=768
-      pipe.enable_attention_slicing()
+      #pipe.enable_attention_slicing()
 
   print(prompt,neg_prompt)