C4G-HKUST commited on
Commit
ae6ee70
·
1 Parent(s): a3f3eee

fix: debug

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -598,7 +598,7 @@ def run_graio_demo(args):
598
  # 使用 @spaces.GPU 装饰器包装 generate_video 函数(参考 LivePortrait)
599
  # 参考: https://huggingface.co/spaces/KlingTeam/LivePortrait/blob/main/app.py
600
  # @spaces.GPU 装饰器会自动处理 GPU 初始化,不需要手动初始化
601
- @spaces.GPU(duration=720)
602
  def gpu_wrapped_generate_video(*args, **kwargs):
603
  # 在 worker 进程中将模型移动到 GPU(如果模型在 CPU 上)
604
  # 参考 LivePortrait: 在 worker 进程中直接使用 .to("cuda")
 
598
  # 使用 @spaces.GPU 装饰器包装 generate_video 函数(参考 LivePortrait)
599
  # 参考: https://huggingface.co/spaces/KlingTeam/LivePortrait/blob/main/app.py
600
  # @spaces.GPU 装饰器会自动处理 GPU 初始化,不需要手动初始化
601
+ @spaces.GPU(duration=120)
602
  def gpu_wrapped_generate_video(*args, **kwargs):
603
  # 在 worker 进程中将模型移动到 GPU(如果模型在 CPU 上)
604
  # 参考 LivePortrait: 在 worker 进程中直接使用 .to("cuda")