upd

app.py

@@ -12,7 +12,9 @@ from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDisc
 model_id = "timbrooks/instruct-pix2pix"
 pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16") #, safety_checker=None)
 pipe.to("cuda")
-pipe.enable_attention_slicing()
+#pipe.enable_attention_slicing()
+pipe.enable_xformers_memory_efficient_attention()
+pipe.unet.to(memory_format=torch.channels_last)
 
 counter = 0