adapt for bi100

2025-09-11 13:50:28 +08:00
parent 2888ee8043
commit cc00e13198
21 changed files with 9 additions and 130 deletions
--- a/test.sh
+++ b/test.sh
@@ -1 +1,3 @@
-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 
+#! /usr/bin/env bash
+
+python3 main.py --model "/mnt/contest_ceph/zhanghao/models/AI-ModelScope/text-to-video-ms-1.7b" --json  "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp32 --model_type text-to-video-ms