C4G-HKUST commited on
Commit
c2e9b28
·
1 Parent(s): ff7f08b

feat: time out check

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -682,8 +682,8 @@ def run_graio_demo(args):
682
 
683
  return gpu_wrapped_generate_video_worker(*args, **kwargs)
684
 
685
- # 高质量生成模式:780秒,用户选择去噪步数
686
- @spaces.GPU(duration=780)
687
  def gpu_wrapped_generate_video_quality(*args, **kwargs):
688
  return gpu_wrapped_generate_video_worker(*args, **kwargs)
689
 
@@ -842,16 +842,16 @@ def run_graio_demo(args):
842
  scale=1
843
  )
844
  run_i2v_button_quality = gr.Button(
845
- "Generate Video (Quality - 780s, Custom steps)",
846
  variant="primary",
847
  scale=1
848
  )
849
  gr.Markdown("""
850
  **Generation Modes:**
851
  - **Fast Mode (up to 150s GPU budget)**: Fixed 8 denoising steps for quick generation. **⚠️ Maximum video duration: 5 seconds. Videos longer than 5 seconds will be automatically trimmed to 5 seconds.**
852
- - **Quality Mode (up to 780s GPU budget)**: Custom denoising steps (adjustable via "Diffusion steps" slider, default: 25 steps). **⚠️ Maximum video duration: 8 seconds with default 25 denoising steps.**
853
 
854
- *Note: The GPU duration (150s/780s) represents the maximum budget allocated, not the actual generation time. Multi-person videos generally require longer duration and more Usage Quota for better quality.*
855
  """)
856
 
857
  with gr.Column(scale=2):
@@ -943,7 +943,7 @@ def run_graio_demo(args):
943
  outputs=[result_gallery],
944
  )
945
 
946
- # 高质量生成按钮:780秒,用户选择步数
947
  run_i2v_button_quality.click(
948
  fn=handle_quality_generation,
949
  inputs=[img2vid_image, img2vid_prompt, n_prompt, img2vid_audio_1, img2vid_audio_2, img2vid_audio_3, sd_steps, seed, guide_scale, person_num_selector, audio_mode_selector],
 
682
 
683
  return gpu_wrapped_generate_video_worker(*args, **kwargs)
684
 
685
+ # 高质量生成模式:720秒,用户选择去噪步数
686
+ @spaces.GPU(duration=720)
687
  def gpu_wrapped_generate_video_quality(*args, **kwargs):
688
  return gpu_wrapped_generate_video_worker(*args, **kwargs)
689
 
 
842
  scale=1
843
  )
844
  run_i2v_button_quality = gr.Button(
845
+ "Generate Video (Quality - 720s, Custom steps)",
846
  variant="primary",
847
  scale=1
848
  )
849
  gr.Markdown("""
850
  **Generation Modes:**
851
  - **Fast Mode (up to 150s GPU budget)**: Fixed 8 denoising steps for quick generation. **⚠️ Maximum video duration: 5 seconds. Videos longer than 5 seconds will be automatically trimmed to 5 seconds.**
852
+ - **Quality Mode (up to 720s GPU budget)**: Custom denoising steps (adjustable via "Diffusion steps" slider, default: 25 steps).
853
 
854
+ *Note: The GPU duration (150s/720s) represents the maximum budget allocated, not the actual generation time. Multi-person videos generally require longer duration and more Usage Quota for better quality.*
855
  """)
856
 
857
  with gr.Column(scale=2):
 
943
  outputs=[result_gallery],
944
  )
945
 
946
+ # 高质量生成按钮:720秒,用户选择步数
947
  run_i2v_button_quality.click(
948
  fn=handle_quality_generation,
949
  inputs=[img2vid_image, img2vid_prompt, n_prompt, img2vid_audio_1, img2vid_audio_2, img2vid_audio_3, sd_steps, seed, guide_scale, person_num_selector, audio_mode_selector],