C4G-HKUST commited on
Commit
b87c83f
·
1 Parent(s): aee5297

feat: time out check

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -618,8 +618,8 @@ def run_graio_demo(args):
618
  # 参考: https://huggingface.co/spaces/KlingTeam/LivePortrait/blob/main/app.py
619
  # @spaces.GPU 装饰器会自动处理 GPU 初始化,不需要手动初始化
620
 
621
- # 快速生成模式:220秒,固定10步去噪
622
- @spaces.GPU(duration=220)
623
  def gpu_wrapped_generate_video_fast(*args, **kwargs):
624
  # 固定使用10步去噪,通过关键字参数传递
625
  kwargs['fixed_steps'] = 8
@@ -837,7 +837,7 @@ def run_graio_demo(args):
837
 
838
  with gr.Row():
839
  run_i2v_button_fast = gr.Button(
840
- "Generate Video (Fast - 220s, 8 steps)",
841
  variant="secondary",
842
  scale=1
843
  )
@@ -848,10 +848,10 @@ def run_graio_demo(args):
848
  )
849
  gr.Markdown("""
850
  **Generation Modes:**
851
- - **Fast Mode (up to 220s GPU budget)**: Fixed 8 denoising steps for quick generation.
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 (220s/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):
@@ -936,7 +936,7 @@ def run_graio_demo(args):
936
  result = gpu_wrapped_generate_video_quality(*args)
937
  return result
938
 
939
- # 快速生成按钮:220秒,固定10步
940
  run_i2v_button_fast.click(
941
  fn=handle_fast_generation,
942
  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],
 
618
  # 参考: https://huggingface.co/spaces/KlingTeam/LivePortrait/blob/main/app.py
619
  # @spaces.GPU 装饰器会自动处理 GPU 初始化,不需要手动初始化
620
 
621
+ # 快速生成模式:100秒,固定10步去噪
622
+ @spaces.GPU(duration=100)
623
  def gpu_wrapped_generate_video_fast(*args, **kwargs):
624
  # 固定使用10步去噪,通过关键字参数传递
625
  kwargs['fixed_steps'] = 8
 
837
 
838
  with gr.Row():
839
  run_i2v_button_fast = gr.Button(
840
+ "Generate Video (Fast - 100s, 8 steps)",
841
  variant="secondary",
842
  scale=1
843
  )
 
848
  )
849
  gr.Markdown("""
850
  **Generation Modes:**
851
+ - **Fast Mode (up to 100s GPU budget)**: Fixed 8 denoising steps for quick generation.
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 (100s/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):
 
936
  result = gpu_wrapped_generate_video_quality(*args)
937
  return result
938
 
939
+ # 快速生成按钮:100秒,固定10步
940
  run_i2v_button_fast.click(
941
  fn=handle_fast_generation,
942
  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],