transformers==4.18.0
torch
torchaudio
librosa==0.9.1
note-seq==0.0.3
pretty-midi==0.2.9
einops==0.4.1
# protobuf==3.20
# numpy==1.22
hydra-core==1.2.0
# typing_extensions
pytorch-lightning
wandb
immutabledict
timm
matplotlib
mir_eval