From 08b5a1433abcf034ad55e834fdbeb462ea7bd8eb Mon Sep 17 00:00:00 2001 From: ZHANG Hao Date: Mon, 8 Sep 2025 12:10:18 +0800 Subject: [PATCH] fix npu iic.py --- iic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/iic.py b/iic.py index 0885768..b968356 100644 --- a/iic.py +++ b/iic.py @@ -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)