flow-matching / environment.yml
sabertoaster's picture
Upload folder using huggingface_hub
4edc9aa verified
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