name: fmri_encoder
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults

dependencies:
  - python=3.10
  - torchaudio=2.5.1
  - pytorch-cuda=12.1
  - torchvision=0.20.1
  - pytorch=2.5.1
  - pip
  - pip:
      - lightning>=2.0.0
      - torchmetrics>=0.11.4
      - hydra-core==1.3.2
      - hydra-colorlog==1.2.0
      - hydra-optuna-sweeper==1.2.0
      - rootutils
      - rich
      - phonemizer
      - tensorboard
      - librosa
      - Cython
      - numpy
      - einops
      - inflect
      - Unidecode
      - scipy
      - matplotlib
      - pandas
      - conformer==0.3.2
      - diffusers
      - seaborn

      # Additional dependencies for fMRI encoding
      - timm
      - h5py
      - omegaconf

prefix: /raid/lttung05/miniconda3/envs/fmri_encoder