Files
enginex-mlu370-text2video/README.md
2025-09-19 10:29:47 +08:00

967 B
Raw Permalink Blame History

Quickstart

构建镜像

docker build -t text2video:v0.1 .

其中基础镜像 mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310 联系寒武纪厂商技术支持可获取

模型下载

模型地址: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目录下生成视频文件。
./run_in_docker.sh

测试结果

A100 平均生成时间(秒) MLU-x4 平均生成时间(秒) MLU-x8 平均生成时间(秒)
时间 12 37 45