fix ascend npu

Merge branch 'feat/ascend' of https://git.modelhub.org.cn:9443/EngineX-Ascend/enginex-ascend-910-text2video into feat/ascend
renaming log
2025-09-16 13:38:17 +08:00 · 2025-09-09 12:59:11 +08:00 · 2025-09-09 12:58:49 +08:00
2 changed files with 2 additions and 2 deletions
--- a/main.py
+++ b/main.py
@@ -147,7 +147,7 @@ def main():

    print(f"\nAll done. vidoes: {len(records)}, total_elapsed: {total_elapsed:.3f}s, avg_latency: {avg_latency:.3f}")
    print(f"Results JSON: {results_path}")
-    print(f"Images dir : {out_dir.resolve()}")
+    print(f"Output dir : {out_dir.resolve()}")


 if __name__ == "__main__":
--- 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
Author	SHA1	Message	Date
ZHANG Hao	45d381d4bb	fix ascend npu	2025-09-16 13:38:17 +08:00
ZHANG Hao	04be0dd34b	Merge branch 'feat/ascend' of https://git.modelhub.org.cn:9443/EngineX-Ascend/enginex-ascend-910-text2video into feat/ascend	2025-09-09 12:59:11 +08:00
yuxiaojie@4paradigm.com	8bd074c118	renaming log	2025-09-09 12:58:49 +08:00