Update app.py

app.py

@@ -21,6 +21,8 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 # need to explicitly call pipe.to("cuda").
 pipe.enable_model_cpu_offload()
 
+pipe.enable_xformers_memory_efficient_attention()
+
 # Generator seed,
 generator = torch.manual_seed(0)
 
@@ -41,7 +43,8 @@ def generate_images(image, prompt):
         prompt,
         canny_image,
         generator=generator,
-        num_images_per_prompt=3
+        num_images_per_prompt=3,
+        num_inference_steps=20,
     )
     all_outputs = []
     all_outputs.append(canny_image)