Gemini899 commited on
Commit
dc2eb09
·
verified ·
1 Parent(s): bca8a55

Update app.py

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