fix ascend npu
This commit is contained in:
2
test.sh
2
test.sh
@@ -1 +1 @@
|
||||
python3 main.py --model "/mnt/contest_ceph/zhanghao/models/iic/text-to-video-synthesis" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16
|
||||
python3 main.py --model "/mnt/contest_ceph/zhanghao/models/iic/text-to-video-synthesis" --json "dataset.json" --results "results.json" --outdir "output" --device npu --dtype fp16
|
||||
|
||||
Reference in New Issue
Block a user