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

Merge branch 'master' of https://gitlab.4pd.io/zhanghao/text2video

Browse Source
This commit is contained in:
yuxiaojie@4paradigm.com
2025-09-09 07:08:24 +08:00
parent cea32765a4 5e2769ecd6
commit 434159b16b
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Dockerfile.a100
View File

@@ -2,6 +2,7 @@ FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
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 -i https://nexus.4pd.io/repository/pypi-all/simple
RUN pip install pytorch-lightning 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 -i https://nexus.4pd.io/repository/pypi-all/simple
RUN pip install transformers accelerate -i https://nexus.4pd.io/repository/pypi-all/simple
COPY . /workspace/

2
run_in_docker_a100.sh
View File

@@ -1,3 +1,3 @@
#! /usr/bin/env bash
image=harbor-contest.4pd.io/zhanghao/t2v:a100-0.1
image=harbor-contest.4pd.io/zhanghao/t2v:a100-0.2
docker run -it -v /home/zhanghao/workspace:/workspace -v /mnt:/mnt $image bash