Spaces:
Sleeping
Sleeping
Tony Lian
commited on
Commit
•
4875c08
1
Parent(s):
fef98ae
Reduce so_batch_size for lower memory cost
Browse files
app.py
CHANGED
@@ -109,7 +109,7 @@ def get_ours_image(response, overall_prompt_override="", seed=0, num_inference_s
|
|
109 |
spec, bg_seed=seed, overall_prompt_override=overall_prompt_override, fg_seed_start=fg_seed_start,
|
110 |
fg_blending_ratio=fg_blending_ratio,frozen_step_ratio=frozen_step_ratio, use_autocast=use_autocast,
|
111 |
gligen_scheduled_sampling_beta=gligen_scheduled_sampling_beta, num_inference_steps=num_inference_steps, scheduler_key=scheduler_key,
|
112 |
-
so_negative_prompt=so_negative_prompt, overall_negative_prompt=overall_negative_prompt, so_batch_size=
|
113 |
)
|
114 |
images = [image_np]
|
115 |
if show_so_imgs:
|
|
|
109 |
spec, bg_seed=seed, overall_prompt_override=overall_prompt_override, fg_seed_start=fg_seed_start,
|
110 |
fg_blending_ratio=fg_blending_ratio,frozen_step_ratio=frozen_step_ratio, use_autocast=use_autocast,
|
111 |
gligen_scheduled_sampling_beta=gligen_scheduled_sampling_beta, num_inference_steps=num_inference_steps, scheduler_key=scheduler_key,
|
112 |
+
so_negative_prompt=so_negative_prompt, overall_negative_prompt=overall_negative_prompt, so_batch_size=2
|
113 |
)
|
114 |
images = [image_np]
|
115 |
if show_so_imgs:
|