Update app.py

app.py

@@ -25,6 +25,7 @@ pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
   safety_checker=None,
   feature_extractor=feature_extractor
 )
+pipe_i2i.enable_xformers_memory_efficient_attention()
 
 if torch.cuda.is_available():
   pipe = pipe.to("cuda")