fix: debug
Browse files
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=
|
| 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")
|