multimodalart HF Staff commited on
Commit
02f82c1
·
verified ·
1 Parent(s): 0fb3ed4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ optimize_pipeline_(
69
  def get_duration(prompt, input_images=None, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=50, guidance_scale=2.5, progress=gr.Progress(track_tqdm=True)):
70
  num_images = 0 if input_images is None else len(input_images)
71
  step_duration = 1 + 0.7 * num_images
72
- return num_inference_steps * step_duration + 10
73
 
74
 
75
  @spaces.GPU(duration=get_duration)
 
69
  def get_duration(prompt, input_images=None, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=50, guidance_scale=2.5, progress=gr.Progress(track_tqdm=True)):
70
  num_images = 0 if input_images is None else len(input_images)
71
  step_duration = 1 + 0.7 * num_images
72
+ return max(65, num_inference_steps * step_duration + 10)
73
 
74
 
75
  @spaces.GPU(duration=get_duration)