Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -161,7 +161,7 @@ def update_dimensions_from_image(image_list):
|
|
| 161 |
def get_duration(prompt_embeds, image_list, width, height, num_inference_steps, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
|
| 162 |
num_images = 0 if image_list is None else len(image_list)
|
| 163 |
step_duration = 1 + 0.8 * num_images
|
| 164 |
-
return max(
|
| 165 |
|
| 166 |
@spaces.GPU(duration=get_duration)
|
| 167 |
def generate_image(prompt_embeds, image_list, width, height, num_inference_steps, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
|
|
|
|
| 161 |
def get_duration(prompt_embeds, image_list, width, height, num_inference_steps, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
|
| 162 |
num_images = 0 if image_list is None else len(image_list)
|
| 163 |
step_duration = 1 + 0.8 * num_images
|
| 164 |
+
return max(35, num_inference_steps * step_duration + 10)
|
| 165 |
|
| 166 |
@spaces.GPU(duration=get_duration)
|
| 167 |
def generate_image(prompt_embeds, image_list, width, height, num_inference_steps, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
|