patrickvonplaten commited on
Commit
e5d785e
1 Parent(s): 724b5bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -22,6 +22,9 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
22
  # need to explicitly call pipe.to("cuda").
23
  pipe.enable_model_cpu_offload()
24
 
 
 
 
25
  # Generator seed,
26
  generator = torch.manual_seed(0)
27
 
@@ -35,7 +38,8 @@ def generate_images(image, prompt):
35
  prompt,
36
  pose,
37
  generator=generator,
38
- num_images_per_prompt=3
 
39
  )
40
  all_outputs = []
41
  all_outputs.append(pose)
 
22
  # need to explicitly call pipe.to("cuda").
23
  pipe.enable_model_cpu_offload()
24
 
25
+ # xformers
26
+ pipe.enable_xformers_memory_efficient_attention()
27
+
28
  # Generator seed,
29
  generator = torch.manual_seed(0)
30
 
 
38
  prompt,
39
  pose,
40
  generator=generator,
41
+ num_images_per_prompt=3,
42
+ num_inference_steps=20,
43
  )
44
  all_outputs = []
45
  all_outputs.append(pose)