Update Dockerfile

Dockerfile

@@ -1,3 +1,5 @@
+# Dockerfile (CORRIGIDO)
+
 # 1. Comece com uma imagem base que já tem o Miniconda
 FROM continuumio/miniconda3
 
@@ -7,29 +9,31 @@ SHELL ["/bin/bash", "-c"]
 # 3. Copie o arquivo de definição do ambiente para dentro do contêiner
 COPY environment.yml .
 
-# 4. Crie o ambiente conda a partir do arquivo. Isso vai instalar tudo.
+# 4. Crie o ambiente conda a partir do arquivo. Agora isso deve funcionar sem erros.
 RUN conda env create -f environment.yml
 
-# 5. Defina o PATH do sistema para usar o Python e os executáveis do nosso ambiente 'seedvr' por padrão.
-# Esta é a parte mais importante, ela substitui a necessidade de 'conda activate'.
+# 5. !!! NOVO PASSO !!!
+#    Agora, instale o flash_attn separadamente DENTRO do ambiente criado.
+#    'conda run -n seedvr' executa o comando no ambiente 'seedvr'.
+RUN conda run -n seedvr pip install "flash_attn==2.5.9.post1" --no-build-isolation
+
+# 6. Defina o PATH do sistema para usar o Python e os executáveis do nosso ambiente 'seedvr' por padrão.
 ENV PATH /opt/conda/envs/seedvr/bin:$PATH
 
-# 6. Crie o diretório do aplicativo
+# 7. Crie o diretório do aplicativo
 WORKDIR /app
 
-# 7. Clone o repositório do SeedVR
+# 8. Clone o repositório do SeedVR
 RUN git clone https://github.com/bytedance-seed/SeedVR.git
 
-# 8. Mude para o diretório do repositório
+# 9. Mude para o diretório do repositório
 WORKDIR /app/SeedVR
 
-# 9. Baixe os checkpoints do modelo durante a construção da imagem para acelerar o início
-# Use o huggingface-cli que foi instalado via pip
+# 10. Baixe os checkpoints do modelo durante a construção da imagem
 RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
 
-# 10. Copie o código do nosso aplicativo Gradio para o diretório de trabalho
-# O app.py deve estar na raiz do seu repositório do Space
+# 11. Copie o código do nosso aplicativo Gradio
 COPY app.py .
 
-# 11. Defina o comando para iniciar o aplicativo Gradio quando o Space for executado
+# 12. Defina o comando para iniciar o aplicativo
 CMD ["python", "app.py"]