Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -77,7 +77,7 @@ def update_dimensions_from_image(image_list):
|
|
| 77 |
def get_duration(prompt_embeds, image_list, width, height, num_inference_steps, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
|
| 78 |
num_images = 0 if image_list is None else len(image_list)
|
| 79 |
step_duration = 1 + 0.8 * num_images
|
| 80 |
-
return max(
|
| 81 |
|
| 82 |
@spaces.GPU(duration=get_duration)
|
| 83 |
def generate_image(prompt_embeds, image_list, width, height, num_inference_steps, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
|
|
|
|
| 77 |
def get_duration(prompt_embeds, image_list, width, height, num_inference_steps, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
|
| 78 |
num_images = 0 if image_list is None else len(image_list)
|
| 79 |
step_duration = 1 + 0.8 * num_images
|
| 80 |
+
return max(35, num_inference_steps * step_duration + 10)
|
| 81 |
|
| 82 |
@spaces.GPU(duration=get_duration)
|
| 83 |
def generate_image(prompt_embeds, image_list, width, height, num_inference_steps, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
|