aducsdr commited on
Commit
767ab4c
·
verified ·
1 Parent(s): 9f65008

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +11 -9
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- # Dockerfile (VERSÃO FINAL COM LIMPEZA DE CACHE)
2
 
3
  # 1. COMEÇAR COM A BASE CORRETA
4
  FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
@@ -27,25 +27,27 @@ RUN conda update -n base -c defaults conda
27
 
28
  # 7. CRIAR O AMBIENTE CONDA
29
  COPY environment.yml .
30
- RUN conda env create -f environment.yml
31
 
32
- # 8. !!! PASSO DE OTIMIZAÇÃO !!! LIMPAR CACHES PARA LIBERAR MEMÓRIA
33
- RUN conda clean --all -y
34
-
35
- # 9. INSTALAR FLASH_ATTN DENTRO DO AMBIENTE
36
  RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation
37
 
38
- # 10. DEFINIR O AMBIENTE CONDA COMO PADRÃO
39
  SHELL ["conda", "run", "-n", "seedvr", "/bin/bash", "-c"]
40
 
41
- # 11. PREPARAR O APLICATIVO
42
  WORKDIR /app
43
  RUN git clone https://github.com/bytedance-seed/SeedVR.git
44
  WORKDIR /app/SeedVR
45
 
46
- # 12. BAIXAR O MODELO DURANTE A CONSTRUÇÃO
47
  RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
48
 
 
 
 
 
 
49
  # 13. COPIAR O CÓDIGO DO NOSSO APP
50
  COPY app.py .
51
 
 
1
+ # Dockerfile (VERSÃO FINAL COM VÍDEOS DE EXEMPLO)
2
 
3
  # 1. COMEÇAR COM A BASE CORRETA
4
  FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
27
 
28
  # 7. CRIAR O AMBIENTE CONDA
29
  COPY environment.yml .
30
+ RUN conda env create -f environment.yml && conda clean --all -y
31
 
32
+ # 8. INSTALAR FLASH_ATTN DENTRO DO AMBIENTE
 
 
 
33
  RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation
34
 
35
+ # 9. DEFINIR O AMBIENTE CONDA COMO PADRÃO
36
  SHELL ["conda", "run", "-n", "seedvr", "/bin/bash", "-c"]
37
 
38
+ # 10. PREPARAR O APLICATIVO
39
  WORKDIR /app
40
  RUN git clone https://github.com/bytedance-seed/SeedVR.git
41
  WORKDIR /app/SeedVR
42
 
43
+ # 11. BAIXAR O MODELO DURANTE A CONSTRUÇÃO
44
  RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
45
 
46
+ # 12. !!! NOVO !!! BAIXAR OS VÍDEOS DE EXEMPLO
47
+ RUN wget -O 01.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/23_1_lq.mp4 && \
48
+ wget -O 02.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/28_1_lq.mp4 && \
49
+ wget -O 03.mp4 https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/2_1_lq.mp4
50
+
51
  # 13. COPIAR O CÓDIGO DO NOSSO APP
52
  COPY app.py .
53