fix npu iic.py

This commit is contained in:
2025-09-08 12:10:18 +08:00
parent db19b64849
commit 08b5a1433a

2
iic.py
View File

@@ -1,11 +1,11 @@
import os
import torch
device = "cuda" if torch.cuda.is_available() else "npu"
import patch
from modelscope.pipelines import pipeline
from modelscope.outputs import OutputKeys
device = "cuda" if torch.cuda.is_available() else "npu"
model_path = "/mnt/contest_ceph/zhanghao/models/iic/text-to-video-synthesis"
p = pipeline('text-to-video-synthesis', model_path, device=device)