Gemini899 commited on
Commit
dfdc124
·
verified ·
1 Parent(s): 540203d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(45, 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)):
 
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)):