386870431bfe6ac37916f4450e6d822ad06201b7
Quickstart
Start with Docker
docker build -t text2video-metax .
metax-docker run -it --gpus=[0] -v <downloaded_model>:<model_dir_in_container> text2video-metax
metax-docker 是沐曦官方的 docker wrapper, 允许自动将 gpu 挂载到容器内. Base 镜像需自行联系沐曦官方获取.
模型下载
模型地址:https://modelscope.cn/models/iic/text-to-video-synthesis
测试程序
- 准备输入数据集,可以参考示例
dataset.json - 在docker镜像里运行测试程序,会根据
dataset.json内容,在output目录下生成视频文件。
python3 main.py --model "<model_dir_in_container>" --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%