aducsdr commited on
Commit
3fe558a
verified
1 Parent(s): d2a4e87

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +9 -17
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- # Dockerfile (VERS脙O FINAL COM LIMITA脟脙O DE JOBS)
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 CORRE脟脙O FINAL !!! INSTALAR FLASH_ATTN USANDO APENAS 1 N脷CLEO
36
- # Definimos MAX_JOBS=1 para for莽ar a compila莽茫o em s茅rie, evitando picos de mem贸ria.
37
- RUN conda run -n seedvr env MAX_JOBS=1 pip install "flash_attn==2.5.9.post1" --no-build-isolation --no-cache-dir
38
 
39
- # 10. BAIXAR OS V脥DEOS DE EXEMPLO
 
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
- # 12. TORNAR O SCRIPT DE INICIALIZA脟脙O EXECUT脕VEL
49
- WORKDIR /app
50
- RUN chmod +x run.sh
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"]