|
|
|
|
|
|
|
|
|
|
|
FROM nvidia/cuda:12.1.1-devel-ubuntu22.04 |
|
|
|
|
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive |
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \ |
|
|
wget \ |
|
|
git \ |
|
|
&& apt-get clean \ |
|
|
&& rm -rf /var/lib/apt/lists/* |
|
|
|
|
|
|
|
|
RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \ |
|
|
/bin/bash ~/miniconda.sh -b -p /opt/conda && \ |
|
|
rm ~/miniconda.sh |
|
|
|
|
|
|
|
|
ENV PATH /opt/conda/bin:$PATH |
|
|
|
|
|
|
|
|
RUN yes | conda tos accept |
|
|
|
|
|
|
|
|
RUN conda update -n base -c defaults conda |
|
|
|
|
|
|
|
|
COPY environment.yml . |
|
|
RUN conda env create -f environment.yml && conda clean --all -y |
|
|
|
|
|
|
|
|
RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation |
|
|
|
|
|
|
|
|
SHELL ["conda", "run", "-n", "seedvr", "/bin/bash", "-c"] |
|
|
|
|
|
|
|
|
WORKDIR /app |
|
|
RUN git clone https://github.com/bytedance-seed/SeedVR.git |
|
|
WORKDIR /app/SeedVR |
|
|
|
|
|
|
|
|
RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False |
|
|
|
|
|
|
|
|
RUN wget -O 01.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/23_1_lq.mp4 && \ |
|
|
wget -O 02.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/28_1_lq.mp4 && \ |
|
|
wget -O 03.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/2_1_lq.mp4 |
|
|
|
|
|
|
|
|
COPY app.py . |
|
|
|
|
|
|
|
|
|
|
|
CMD ["/opt/conda/envs/seedvr/bin/python", "app.py"] |