wenmengzhou commited on
Commit
616413a
·
verified ·
1 Parent(s): fc36676

enlarge duration of zero gpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -212,7 +212,7 @@ with gr.Blocks() as app:
212
  outputs=[output_image],
213
  triggers=run_button.click
214
  )
215
- @spaces.GPU(duration=60)
216
  def generate_image(model_type, model_path, prompt, negative_prompt, cfg_scale, embedded_guidance, num_inference_steps, height, width, seed, *args, progress=gr.Progress()):
217
  _, pipe = load_model(model_type, model_path)
218
  input_params = {
 
212
  outputs=[output_image],
213
  triggers=run_button.click
214
  )
215
+ @spaces.GPU(duration=120)
216
  def generate_image(model_type, model_path, prompt, negative_prompt, cfg_scale, embedded_guidance, num_inference_steps, height, width, seed, *args, progress=gr.Progress()):
217
  _, pipe = load_model(model_type, model_path)
218
  input_params = {