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