diff --git a/Dockerfile.mrv100 b/Dockerfile.mrv100 new file mode 100644 index 0000000..1bcca59 --- /dev/null +++ b/Dockerfile.mrv100 @@ -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/ diff --git a/run_in_docker_mrv100.sh b/run_in_docker_mrv100.sh new file mode 100755 index 0000000..9fb270f --- /dev/null +++ b/run_in_docker_mrv100.sh @@ -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 diff --git a/test_ms.sh b/test_ms.sh index f66d92d..70a2f4e 100755 --- a/test_ms.sh +++ b/test_ms.sh @@ -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"