Update Dockerfile
Browse files- Dockerfile +9 -17
Dockerfile
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
# Dockerfile (VERS脙O FINAL COM
|
| 2 |
|
| 3 |
# 1. COME脟AR COM A BASE CORRETA
|
| 4 |
FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
|
|
@@ -32,23 +32,15 @@ RUN conda update -n base -c defaults conda
|
|
| 32 |
# 8. CRIAR O AMBIENTE CONDA
|
| 33 |
RUN conda env create -f environment.yml && conda clean --all -y
|
| 34 |
|
| 35 |
-
# 9. !!! A
|
| 36 |
-
|
| 37 |
-
|
| 38 |
|
| 39 |
-
# 10. BAIXAR
|
|
|
|
| 40 |
WORKDIR /app/SeedVR
|
| 41 |
-
RUN wget -O 01.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/23_1_lq.mp4 && \
|
| 42 |
-
wget -O 02.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/28_1_lq.mp4 && \
|
| 43 |
-
wget -O 03.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/2_1_lq.mp4
|
| 44 |
-
|
| 45 |
-
# 11. BAIXAR O MODELO
|
| 46 |
RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
|
| 47 |
|
| 48 |
-
#
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
# 13. DEFINIR O PONTO DE ENTRADA
|
| 53 |
-
WORKDIR /app/SeedVR
|
| 54 |
-
CMD ["/app/run.sh"]
|
|
|
|
| 1 |
+
# Dockerfile (VERS脙O FINAL COM SKIP CUDA BUILD)
|
| 2 |
|
| 3 |
# 1. COME脟AR COM A BASE CORRETA
|
| 4 |
FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
|
|
|
|
| 32 |
# 8. CRIAR O AMBIENTE CONDA
|
| 33 |
RUN conda env create -f environment.yml && conda clean --all -y
|
| 34 |
|
| 35 |
+
# 9. !!! A M脕GICA FINAL !!! INSTALAR FLASH_ATTN SEM COMPILAR CUDA
|
| 36 |
+
RUN conda run -n seedvr env FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE \
|
| 37 |
+
pip install "flash_attn==2.5.9.post1" --no-build-isolation --no-cache-dir
|
| 38 |
|
| 39 |
+
# 10. BAIXAR O MODELO
|
| 40 |
+
# (O script de infer锚ncia espera que estejamos no diret贸rio /app/SeedVR)
|
| 41 |
WORKDIR /app/SeedVR
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
|
| 43 |
|
| 44 |
+
# 11. DEFINIR UM COMANDO DE "ESPERA"
|
| 45 |
+
# Este comando simplesmente mant茅m o cont锚iner rodando para que possamos usar o terminal.
|
| 46 |
+
CMD ["tail", "-f", "/dev/null"]
|
|
|
|
|
|
|
|
|
|
|
|