## Install Packages
1. `python3 -m venv env`
2. `source env/bin/activate`
3. `pip install -e .`
4. `pip install -e umx`

## Download [GuitarFX Dataset] (https://zenodo.org/record/7044411/)
`./download_egfx.sh`

## Train model
1. Change Wandb variables in `shell_vars.sh`
2. `python train.py exp=audio_diffusion`
or
2. `python train.py exp=umx`

To add gpu, add `trainer.accelerator='gpu' trainer.devices=-1` to the command-line

Ex. `python train.py exp=umx trainer.accelerator='gpu' trainer.devices=-1`