test

app.py

@@ -21,6 +21,8 @@ model_id = "timbrooks/instruct-pix2pix"
 pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
     model_id, torch_dtype=torch.float16
 ).to("cuda")
+pipe.enable_xformers_memory_efficient_attention()
+pipe.unet.to(memory_format=torch.channels_last)
 
 
 def new_image(prompt, image, img_guidance, guidance, steps):

requirements.txt

@@ -2,4 +2,4 @@ panel
 hvplot
 diffusers
 transformers
-accelerate
+accelerate