Use @spaces.GPU decorator for GPU management (reference LivePortrait)

app.py

@@ -7,6 +7,7 @@ import json
 import warnings
 from datetime import datetime
 import gradio as gr
+import spaces
 warnings.filterwarnings('ignore')
 
 import random
@@ -572,6 +573,12 @@ def run_graio_demo(args):
         logging.info("Finished.")
         return output_file
 
+    # 使用 @spaces.GPU 装饰器包装 generate_video 函数（参考 LivePortrait）
+    # 参考: https://huggingface.co/spaces/KlingTeam/LivePortrait/blob/main/app.py
+    @spaces.GPU(duration=240)
+    def gpu_wrapped_generate_video(*args, **kwargs):
+        return generate_video(*args, **kwargs)
+
     def toggle_audio_inputs(person_num):
         """根据选择的人数显示对应数量的音频输入框"""
         if person_num == "1 Person":
@@ -696,7 +703,7 @@ def run_graio_demo(args):
 
 
         run_i2v_button.click(
-            fn=generate_video,
+            fn=gpu_wrapped_generate_video,  # 使用 GPU 包装函数
             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],
             outputs=[result_gallery],
         )

requirements.txt

@@ -26,6 +26,7 @@ onnxruntime-gpu==1.22.0
 insightface==0.7.3
 transformers==4.52.0
 huggingface_hub
+spaces
 ninja
 # flash_attn 预编译 wheel (torch 2.6 + CUDA 12 + Python 3.10)
 # 参考: https://huggingface.co/spaces/fffiloni/Meigen-MultiTalk/blob/main/requirements.txt