EngineX-Ascend/enginex-ascend-910-text2video
10
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix npu iic.py

Browse Source
This commit is contained in:
ZHANG Hao
2025-09-08 12:10:18 +08:00
parent db19b64849
commit 08b5a1433a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
iic.py
View File

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