adaface-neurips commited on
Commit
798f7db
1 Parent(s): 02cc20b

increase gpu duration to 90 seconds

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ def gen_init_images(uploaded_image_paths, prompt, adaface_id_cfg_scale, out_imag
94
  # Update uploaded_init_img_gallery, update and hide init_img_files, hide init_clear_button_column
95
  return gr.update(value=face_paths, visible=True), gr.update(value=face_paths, visible=False), gr.update(visible=True)
96
 
97
- @spaces.GPU
98
  def generate_image(image_container, uploaded_image_paths, init_img_file_paths, init_img_selected_idx,
99
  init_image_strength, init_image_final_weight,
100
  prompt, negative_prompt, num_steps, video_length, guidance_scale, seed, attn_scale, image_embed_scale,
 
94
  # Update uploaded_init_img_gallery, update and hide init_img_files, hide init_clear_button_column
95
  return gr.update(value=face_paths, visible=True), gr.update(value=face_paths, visible=False), gr.update(visible=True)
96
 
97
+ @spaces.GPU(duration=90)
98
  def generate_image(image_container, uploaded_image_paths, init_img_file_paths, init_img_selected_idx,
99
  init_image_strength, init_image_final_weight,
100
  prompt, negative_prompt, num_steps, video_length, guidance_scale, seed, attn_scale, image_embed_scale,