fix npu iic.py
This commit is contained in:
2
iic.py
2
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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user