Spaces:
Running
on
Zero
Running
on
Zero
fix: quota
Browse files
app.py
CHANGED
|
@@ -290,7 +290,7 @@ def _parse_args():
|
|
| 290 |
parser.add_argument(
|
| 291 |
"--det_thresh",
|
| 292 |
type=float,
|
| 293 |
-
default=0.
|
| 294 |
help="Threshold for InsightFace face detection.")
|
| 295 |
parser.add_argument(
|
| 296 |
"--mode",
|
|
@@ -605,11 +605,11 @@ def run_graio_demo(args):
|
|
| 605 |
# 参考: https://huggingface.co/spaces/KlingTeam/LivePortrait/blob/main/app.py
|
| 606 |
# @spaces.GPU 装饰器会自动处理 GPU 初始化,不需要手动初始化
|
| 607 |
|
| 608 |
-
# 快速生成模式:
|
| 609 |
-
@spaces.GPU(duration=
|
| 610 |
def gpu_wrapped_generate_video_fast(*args, **kwargs):
|
| 611 |
-
# 固定使用
|
| 612 |
-
kwargs['fixed_steps'] =
|
| 613 |
return gpu_wrapped_generate_video_worker(*args, **kwargs)
|
| 614 |
|
| 615 |
# 高质量生成模式:720秒,用户选择去噪步数
|
|
@@ -757,7 +757,7 @@ def run_graio_demo(args):
|
|
| 757 |
|
| 758 |
with gr.Row():
|
| 759 |
run_i2v_button_fast = gr.Button(
|
| 760 |
-
"Generate Video (Fast -
|
| 761 |
variant="secondary",
|
| 762 |
scale=1
|
| 763 |
)
|
|
@@ -768,10 +768,10 @@ def run_graio_demo(args):
|
|
| 768 |
)
|
| 769 |
gr.Markdown("""
|
| 770 |
**Generation Modes:**
|
| 771 |
-
- **Fast Mode (up to
|
| 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. The 720s is the maximum GPU allocation time, not the actual generation time. With 40 denoising steps, approximately 10 seconds of video can be generated.
|
| 773 |
|
| 774 |
-
*Note: The GPU duration (
|
| 775 |
""")
|
| 776 |
|
| 777 |
with gr.Column(scale=2):
|
|
@@ -806,7 +806,7 @@ def run_graio_demo(args):
|
|
| 806 |
)
|
| 807 |
|
| 808 |
|
| 809 |
-
# 快速生成按钮:
|
| 810 |
run_i2v_button_fast.click(
|
| 811 |
fn=gpu_wrapped_generate_video_fast,
|
| 812 |
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],
|
|
|
|
| 290 |
parser.add_argument(
|
| 291 |
"--det_thresh",
|
| 292 |
type=float,
|
| 293 |
+
default=0.10,
|
| 294 |
help="Threshold for InsightFace face detection.")
|
| 295 |
parser.add_argument(
|
| 296 |
"--mode",
|
|
|
|
| 605 |
# 参考: https://huggingface.co/spaces/KlingTeam/LivePortrait/blob/main/app.py
|
| 606 |
# @spaces.GPU 装饰器会自动处理 GPU 初始化,不需要手动初始化
|
| 607 |
|
| 608 |
+
# 快速生成模式:210秒,固定10步去噪
|
| 609 |
+
@spaces.GPU(duration=210)
|
| 610 |
def gpu_wrapped_generate_video_fast(*args, **kwargs):
|
| 611 |
+
# 固定使用10步去噪,通过关键字参数传递
|
| 612 |
+
kwargs['fixed_steps'] = 10
|
| 613 |
return gpu_wrapped_generate_video_worker(*args, **kwargs)
|
| 614 |
|
| 615 |
# 高质量生成模式:720秒,用户选择去噪步数
|
|
|
|
| 757 |
|
| 758 |
with gr.Row():
|
| 759 |
run_i2v_button_fast = gr.Button(
|
| 760 |
+
"Generate Video (Fast - 210s, 10 steps)",
|
| 761 |
variant="secondary",
|
| 762 |
scale=1
|
| 763 |
)
|
|
|
|
| 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. The 210s is the maximum GPU allocation time, not the actual generation time.
|
| 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. The 720s is the maximum GPU allocation time, not the actual generation time. 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):
|
|
|
|
| 806 |
)
|
| 807 |
|
| 808 |
|
| 809 |
+
# 快速生成按钮:210秒,固定10步
|
| 810 |
run_i2v_button_fast.click(
|
| 811 |
fn=gpu_wrapped_generate_video_fast,
|
| 812 |
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],
|