# 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