EngineX-Iluvatar/enginex-bi_series-text2video
15
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

add dockerfile for mrv100

Browse Source
This commit is contained in:
Zhang Hao
2025-09-09 14:57:59 +08:00
parent 54ac219d77
commit 2888ee8043
3 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
Dockerfile.mrv100 Normal file
View File

@@ -0,0 +1,7 @@
FROM git.modelhub.org.cn:980/enginex-iluvatar/mr100_corex:4.3.0
WORKDIR /workspace
RUN pip install opencv-python-headless imageio[ffmpeg] einops datasets==3.2.0 simplejson diffusers==0.34.0 open_clip_torch==2.24.0 sortedcontainers modelscope av==11.0.0 addict pytorch-lightning
COPY . /workspace/

3
run_in_docker_mrv100.sh Executable file
View File

@@ -0,0 +1,3 @@
#! /usr/bin/env bash
image=harbor-contest.4pd.io/zhanghao/t2v:mrv100-0.1
docker run -it -v `pwd`:/workspace -v /root/models:/root/models -v /mnt:/mnt --device=dev/iluvatar0:/dev/iluvatar0 $image bash

2
test_ms.sh
View File

@@ -1 +1 @@
python3 main.py --model "/mnt/models/AI-ModelScope/text-to-video-ms-1.7b" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp32 --model_type "text-to-video-ms"
python3 main.py --model "/mnt/contest_ceph/zhanghao/models/AI-ModelScope/text-to-video-ms-1.7b" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp32 --model_type "text-to-video-ms"