95a0dea0667c13c8409a477d91bb67d5364e9580
Quickstart
Start with Docker
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 是沐曦官方的 docker wrapper, 允许自动将 gpu 挂载到容器内.
模型下载
模型地址: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%