update README
This commit is contained in:
10
README.md
10
README.md
@@ -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
|
||||
```
|
||||
|
||||
## 测试结果
|
||||
|
||||
Reference in New Issue
Block a user