update README

This commit is contained in:
qiliguo
2025-09-16 18:37:58 +08:00
parent 4ec3a87b14
commit 95a0dea066

View File

@@ -1,19 +1,21 @@
## Quickstart
### 构建镜像
### Start with Docker
```bash
docker build -t text2video:v0.1 .
metax-docker run -it --gpus=[0] -v <downloaded_model>:<model_dir_in_container> git.modelhub.org.cn:9443/enginex-metax/text2video:0.1
```
[metax-docker](https://developer.metax-tech.com/softnova/category?package_kind=Cloud&dimension=metax&chip_name=%E6%9B%A6%E4%BA%91C500%E7%B3%BB%E5%88%97&deliver_type=%E5%88%86%E5%B1%82%E5%8C%85&series_name=metax-docker) 是沐曦官方的 docker wrapper, 允许自动将 gpu 挂载到容器内.
### 模型下载
模型地址https://modelscope.cn/models/iic/text-to-video-synthesis
并放到目录:`/mnt/contest_ceph/zhanghao/models/iic/text-to-video-synthesis`(如更改目录,请修改后面的执行脚本中的模型路径)
### 测试程序
1. 准备输入数据集,可以参考示例`dataset.json`
2. 在docker镜像里运行测试程序会根据`dataset.json`内容,在`output`目录下生成视频文件。
```bash
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 "<model_dir_in_container>" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16
```
## 测试结果