From 45d381d4bb5d3cbfbcc25fa25d05b758541668c2 Mon Sep 17 00:00:00 2001
From: ZHANG Hao <zhanghao@4paradigm.com>
Date: Tue, 16 Sep 2025 13:38:17 +0800
Subject: [PATCH] fix ascend npu

---
 test.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test.sh b/test.sh
index e342652..614f2fb 100755
--- a/test.sh
+++ b/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