Spaces:
Running
Running
Commit
•
e5d785e
1
Parent(s):
724b5bd
Update app.py
Browse files
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)
|