Update app.py

app.py

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