adapt for mr100

2025-09-11 14:03:24 +08:00
parent 07779d756f
commit 08b54c942e
22 changed files with 10 additions and 148 deletions
--- a/test.sh
+++ b/test.sh
@@ -1 +1,4 @@
-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
+
+mkdir output
+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