C4G-HKUST commited on
Commit
7f024cb
·
1 Parent(s): 1dc4720

fix: budget

Browse files
Files changed (1) hide show
  1. app.py +17 -8
app.py CHANGED
@@ -612,8 +612,8 @@ def run_graio_demo(args):
612
  kwargs['fixed_steps'] = 10
613
  return gpu_wrapped_generate_video_worker(*args, **kwargs)
614
 
615
- # 高质量生成模式:720秒,用户选择去噪步数
616
- @spaces.GPU(duration=720)
617
  def gpu_wrapped_generate_video_quality(*args, **kwargs):
618
  return gpu_wrapped_generate_video_worker(*args, **kwargs)
619
 
@@ -684,6 +684,15 @@ def run_graio_demo(args):
684
  <a href='https://huggingface.co/zzz66/AnyTalker-1.3B'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20HuggingFace-Model-yellow'></a>
685
  <a href='https://arxiv.org/abs/2511.23475/'><img src='https://img.shields.io/badge/Paper-Arxiv-red'></a>
686
  </div>
 
 
 
 
 
 
 
 
 
687
 
688
 
689
  """)
@@ -733,7 +742,7 @@ def run_graio_demo(args):
733
  label="Diffusion steps",
734
  minimum=1,
735
  maximum=1000,
736
- value=40,
737
  step=1)
738
  seed = gr.Slider(
739
  label="Seed",
@@ -762,16 +771,16 @@ def run_graio_demo(args):
762
  scale=1
763
  )
764
  run_i2v_button_quality = gr.Button(
765
- "Generate Video (Quality - 720s, Custom steps)",
766
  variant="primary",
767
  scale=1
768
  )
769
  gr.Markdown("""
770
  **Generation Modes:**
771
- - **Fast Mode (up to 210s GPU budget)**: Fixed 10 denoising steps for quick generation. Suitable for single-person videos or quick previews. **For free-tier users: Fast mode can generate approximately 6 seconds of two-person video at most; longer videos may timeout.**
772
- - **Quality Mode (up to 720s GPU budget)**: Custom denoising steps (adjustable via "Diffusion steps" slider). Recommended for multi-person videos that require higher quality. With 40 denoising steps, approximately 10 seconds of video can be generated.
773
 
774
- *Note: The GPU duration (210s/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.*
775
  """)
776
 
777
  with gr.Column(scale=2):
@@ -813,7 +822,7 @@ def run_graio_demo(args):
813
  outputs=[result_gallery],
814
  )
815
 
816
- # 高质量生成按钮:720秒,用户选择步数
817
  run_i2v_button_quality.click(
818
  fn=gpu_wrapped_generate_video_quality,
819
  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],
 
612
  kwargs['fixed_steps'] = 10
613
  return gpu_wrapped_generate_video_worker(*args, **kwargs)
614
 
615
+ # 高质量生成模式:780秒,用户选择去噪步数
616
+ @spaces.GPU(duration=780)
617
  def gpu_wrapped_generate_video_quality(*args, **kwargs):
618
  return gpu_wrapped_generate_video_worker(*args, **kwargs)
619
 
 
684
  <a href='https://huggingface.co/zzz66/AnyTalker-1.3B'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20HuggingFace-Model-yellow'></a>
685
  <a href='https://arxiv.org/abs/2511.23475/'><img src='https://img.shields.io/badge/Paper-Arxiv-red'></a>
686
  </div>
687
+ <div style="background-color: #fff3cd; border: 2px solid #ffc107; border-radius: 8px; padding: 15px; margin: 20px 0; text-align: center;">
688
+ <div style="font-size: 18px; font-weight: bold; color: #856404; margin-bottom: 10px;">
689
+ ⚠️ Important Video Duration Limits
690
+ </div>
691
+ <div style="font-size: 14px; color: #856404; line-height: 1.6;">
692
+ <strong>Fast Mode:</strong> Maximum video duration is <strong>6 seconds</strong>. Videos longer than 6 seconds will timeout.<br>
693
+ <strong>Quality Mode:</strong> Maximum video duration is <strong>8 seconds</strong> with default 25 denoising steps.
694
+ </div>
695
+ </div>
696
 
697
 
698
  """)
 
742
  label="Diffusion steps",
743
  minimum=1,
744
  maximum=1000,
745
+ value=25,
746
  step=1)
747
  seed = gr.Slider(
748
  label="Seed",
 
771
  scale=1
772
  )
773
  run_i2v_button_quality = gr.Button(
774
+ "Generate Video (Quality - 780s, Custom steps)",
775
  variant="primary",
776
  scale=1
777
  )
778
  gr.Markdown("""
779
  **Generation Modes:**
780
+ - **Fast Mode (up to 210s GPU budget)**: Fixed 10 denoising steps for quick generation. **⚠️ Maximum video duration: 6 seconds. Videos longer than 6 seconds will timeout.**
781
+ - **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.**
782
 
783
+ *Note: The GPU duration (210s/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.*
784
  """)
785
 
786
  with gr.Column(scale=2):
 
822
  outputs=[result_gallery],
823
  )
824
 
825
+ # 高质量生成按钮:780秒,用户选择步数
826
  run_i2v_button_quality.click(
827
  fn=gpu_wrapped_generate_video_quality,
828
  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],