4ec3a87b140f11e7aa411141cb2b3c4a2b51ecb1
Quickstart
构建镜像
docker build -t text2video:v0.1 .
模型下载
模型地址:https://modelscope.cn/models/iic/text-to-video-synthesis
并放到目录:/mnt/contest_ceph/zhanghao/models/iic/text-to-video-synthesis(如更改目录,请修改后面的执行脚本中的模型路径)
测试程序
- 准备输入数据集,可以参考示例
dataset.json - 在docker镜像里运行测试程序,会根据
dataset.json内容,在output目录下生成视频文件。
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
测试结果
| A100 平均生成时间(秒) | MetaX C500 平均生成时间(秒) | |
|---|---|---|
| 时间 | 12 | 16 |
Description
Languages
Python
96.5%
Dockerfile
3.5%