SeedVR2 / environment.yml
aducsdr's picture
Update environment.yml
aef4a65 verified
# Nome do ambiente Conda que será criado
name: seedvr
# Canais de onde o Conda irá baixar os pacotes, em ordem de prioridade.
# pytorch e nvidia são essenciais para obter as versões corretas com suporte a GPU.
channels:
- pytorch
- nvidia
- defaults
# Lista de todas as dependências
dependencies:
# Pacotes principais instalados via Conda para máxima compatibilidade com a GPU
- python
- pytorch
- torchvision
- torchaudio
- pytorch-cuda # Garante que o PyTorch seja compatível com a CUDA da imagem Docker
- pip # Instala o gerenciador de pacotes pip dentro do ambiente
# Pacotes instalados via Pip
- pip:
# Dependências da API FastAPI
- fastapi
- uvicorn
- python-multipart
# Dependências do projeto SeedVR
- accelerate
- av
- beautifulsoup4
- controlnet_aux
- diffusers
- einops
- ftfy
- huggingface_hub
- iopath
- mediapy
- omegaconf
- opencv-python
- pandas
- pytorch-lightning
- pytorchvideo
- regex
- scikit-image
- timm
- tqdm
- transformers
- webdataset
- xformers
# NOTA IMPORTANTE:
# 'flash_attn' não está listado aqui de propósito.
# Sua instalação requer flags especiais para evitar erros de memória durante a construção.
# Por isso, ele é instalado separadamente com um comando RUN dedicado no Dockerfile. # A linha do flash_attn foi REMOVIDA daqui