C4G-HKUST commited on
Commit
968cdd5
·
1 Parent(s): 8216d98

Update quality mode duration calculation: change coefficient from 2.5s to 3s per step

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +4 -4
README.md CHANGED
@@ -220,7 +220,7 @@ The Gradio demo provides two generation modes:
220
  - **Quality Mode (Dynamic GPU budget)**:
221
  - Custom denoising steps (adjustable via "Diffusion steps" slider, default: 25 steps)
222
  - Recommended for multi-person videos that require higher quality
223
- - GPU duration is dynamically calculated as: **60s (preprocessing) + video_seconds × steps × 2.5s**
224
  - Longer generation time but better quality output
225
 
226
  **Design Rationale**: Multi-person videos generally have longer duration and require more computational resources. To achieve better quality, especially for complex multi-person interactions, more denoising steps and longer GPU allocation time are needed. The Quality Mode dynamically allocates GPU time based on video length and denoising steps, while the Fast Mode offers a quick preview option with fixed 8 steps and a 4-second maximum duration for faster iteration.
 
220
  - **Quality Mode (Dynamic GPU budget)**:
221
  - Custom denoising steps (adjustable via "Diffusion steps" slider, default: 25 steps)
222
  - Recommended for multi-person videos that require higher quality
223
+ - GPU duration is dynamically calculated as: **60s (preprocessing) + video_seconds × steps × 3s**
224
  - Longer generation time but better quality output
225
 
226
  **Design Rationale**: Multi-person videos generally have longer duration and require more computational resources. To achieve better quality, especially for complex multi-person interactions, more denoising steps and longer GPU allocation time are needed. The Quality Mode dynamically allocates GPU time based on video length and denoising steps, while the Fast Mode offers a quick preview option with fixed 8 steps and a 4-second maximum duration for faster iteration.
app.py CHANGED
@@ -648,9 +648,9 @@ def run_graio_demo(args):
648
  def get_duration(video_seconds, steps):
649
  """
650
  计算quality模式所需的GPU duration
651
- duration = 60s (预处理时间) + 视频秒数 * 步数 * 2.5
652
  """
653
- return int(60 + video_seconds * steps * 2.5)
654
 
655
  # 为quality模式创建动态duration计算函数
656
  def calculate_quality_duration(*args, **kwargs):
@@ -932,7 +932,7 @@ def run_graio_demo(args):
932
  gr.Markdown("""
933
  **Generation Modes:**
934
  - **Fast Mode (120s GPU budget, suitable for any type of users)**: Fixed 8 denoising steps for quick generation. Maximum video duration: 4 seconds.
935
- - **Quality Mode (Dynamic GPU budget)**: Custom denoising steps (adjustable via "Diffusion steps" slider, default: 25 steps). GPU duration is dynamically calculated as: 60s (preprocessing) + video_seconds × steps × 2.5 s.
936
 
937
  *Note: Fast mode has a fixed 120s GPU budget. Quality mode dynamically allocates GPU time based on video length and denoising steps. Multi-person videos generally require longer duration and more Usage Quota for better quality.*
938
  """)
@@ -1058,7 +1058,7 @@ def run_graio_demo(args):
1058
  # 计算实际使用的duration
1059
  actual_duration = get_duration(actual_video_seconds, actual_steps)
1060
  # 使用 gr.Info 提示用户
1061
- info_msg = f"Video generation completed! Duration used: {actual_duration}s (60s preprocessing + {actual_video_seconds:.2f}s video × {actual_steps} steps × 2.5s)"
1062
  gr.Info(info_msg)
1063
  return output_file
1064
  else:
 
648
  def get_duration(video_seconds, steps):
649
  """
650
  计算quality模式所需的GPU duration
651
+ duration = 60s (预处理时间) + 视频秒数 * 步数 * 3
652
  """
653
+ return int(60 + video_seconds * steps * 3)
654
 
655
  # 为quality模式创建动态duration计算函数
656
  def calculate_quality_duration(*args, **kwargs):
 
932
  gr.Markdown("""
933
  **Generation Modes:**
934
  - **Fast Mode (120s GPU budget, suitable for any type of users)**: Fixed 8 denoising steps for quick generation. Maximum video duration: 4 seconds.
935
+ - **Quality Mode (Dynamic GPU budget)**: Custom denoising steps (adjustable via "Diffusion steps" slider, default: 25 steps). GPU duration is dynamically calculated as: 60s (preprocessing) + video_seconds × steps × 3 s.
936
 
937
  *Note: Fast mode has a fixed 120s GPU budget. Quality mode dynamically allocates GPU time based on video length and denoising steps. Multi-person videos generally require longer duration and more Usage Quota for better quality.*
938
  """)
 
1058
  # 计算实际使用的duration
1059
  actual_duration = get_duration(actual_video_seconds, actual_steps)
1060
  # 使用 gr.Info 提示用户
1061
+ info_msg = f"Video generation completed! Duration used: {actual_duration}s (60s preprocessing + {actual_video_seconds:.2f}s video × {actual_steps} steps × 3s)"
1062
  gr.Info(info_msg)
1063
  return output_file
1064
  else: