Spaces:
Running
on
T4
Running
on
T4
File size: 2,579 Bytes
99122c4 50f034f 99122c4 50f034f 99122c4 50f034f 99122c4 50f034f 9da46f9 99122c4 50f034f 9da46f9 50f034f a84c25c 50f034f 99122c4 50f034f 9da46f9 50f034f e3ca5f7 99122c4 50f034f 6fcf6a4 6f55a79 6fcf6a4 99122c4 50f034f 6fcf6a4 99122c4 50f034f 99122c4 50f034f 6fcf6a4 50f034f 99122c4 50f034f 99122c4 50f034f 9da46f9 99122c4 75a7169 9da46f9 75a7169 50f034f 75a7169 50f034f 99122c4 75a7169 e3ca5f7 99122c4 9da46f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
# VampNet
This repository contains recipes for training generative music models on top of the Lyrebird Audio Codec.
# Setting up
install AudioTools
```bash
git clone https://github.com/descriptinc/audiotools.git
pip install -e ./audiotools
```
install the Descript Audio Codec.
```bash
git clone https://github.com/descriptinc/descript-audio-codec.git
pip install -e ./descript-audio-codec
```
install VampNet
```bash
git clone https://github.com/hugofloresgarcia/vampnet.git
pip install -e ./vampnet
```
## A note on argbind
This repository relies on [argbind](https://github.com/pseeth/argbind) to manage CLIs and config files.
Config files are stored in the `conf/` folder.
## Getting the Pretrained Models
Download the pretrained models from [this link](https://drive.google.com/file/d/1ZIBMJMt8QRE8MYYGjg4lH7v7BLbZneq2/view?usp=sharing). Then, extract the models to the `models/` folder.
# Usage
First, you'll want to set up your environment
```bash
source ./env/env.sh
```
## Staging a Run
Staging a run makes a copy of all the git-tracked files in the codebase and saves them to a folder for reproducibility. You can then run the training script from the staged folder.
```
stage --name my_run --run_dir /path/to/staging/folder
```
## Training a model
```bash
python scripts/exp/train.py --args.load conf/vampnet.yml --save_path /path/to/checkpoints
```
See `python scripts/exp/train.py -h` for a list of options.
## Fine-tuning
To fine-tune a model, use the script in `scripts/exp/fine_tune.py` to generate 3 configuration files: `c2f.yml`, `coarse.yml`, and `interface.yml`.
The first two are used to fine-tune the coarse and fine models, respectively. The last one is used to launch the gradio interface.
```bash
python scripts/exp/fine_tune.py "/path/to/audio1.mp3 /path/to/audio2/ /path/to/audio3.wav" <fine_tune_name>
```
This will create a folder under `conf/<fine_tune_name>/` with the 3 configuration files.
The save_paths will be set to `runs/<fine_tune_name>/coarse` and `runs/<fine_tune_name>/c2f`.
launch the coarse job:
```bash
python scripts/exp/train.py --args.load conf/<fine_tune_name>/coarse.yml
```
this will save the coarse model to `runs/<fine_tune_name>/coarse/ckpt/best/`.
launch the c2f job:
```bash
python scripts/exp/train.py --args.load conf/<fine_tune_name>/c2f.yml
```
launch the interface:
```bash
python demo.py --args.load conf/generated/<fine_tune_name>/interface.yml
```
## Launching the Gradio Interface
```bash
python demo.py --args.load conf/interface/spotdl.yml --Interface.device cuda
```
|