This repository contains the code necessary to train Koe AI's LLVC models and to reproduce the LLVC paper.
LLVC paper: https://koe.ai/papers/llvc.pdf
LLVC samples: https://koeai.github.io/llvc-demo/
Windows executable: https://koe.ai/recast/download/
Koe AI homepage: https://koe.ai/
- Create a Python environment with e.g. conda:
conda create -n llvc python=3.11
- Activate the new environment:
conda activate llvc
- Install torch and torchaudio from https://pytorch.org/get-started/locally/
- Install requirements with
pip install -r requirements.txt
- Download models with
python download_models.py
eval.py
has requirements that conflict withrequirements.txt
, so before running this file, create a seperate new Python virtual environment with python 3.9 and runpip install -r eval_requirements.txt
You should now be able to run python infer.py
and convert all of the files in test_wavs
with the pretrained llvc checkpoint, with the resulting files saved to converted_out
.
python infer.py -p my_checkpoint.pth -c my_config.pth -f input_file -o my_out_dir
will convert a single audio file or folder of audio files using the given LLVC checkpoint and save the output to the folder my_out_dir
. The -s
argument simulate a streaming environment for conversion. The -n
argument allows the user to specify the size of input audio chunks in streaming mode, trading increased latency for better RTF.
compare_infer.py
allows you to reproduce our streaming no-f0 RVC and QuickVC conversions on input audio of your choice. By default, window_ms
and extra_convert_size
are set to the values used for no-f0 RVC conversion. See the linked paper for the QuickVC conversion parameters.
- Create a folder
experiments/my_run
containing aconfig.json
(seeexperiments/llvc/config.json
for an example) - Edit the
config.json
to reflect the location of your dataset and desired architectural modifications python train.py -d experiments/my_run
- The run will be logged to Tensorboard in the directory
experiments/my_run/logs
Datasets are comprised of a folder containing three subfolders: dev
, train
and val
. Each of these folders contains audio files of the form PREFIX_original.wav
, which are audio clips recorded by a variety of input speakers, and PREFIX_converted.wav
, which are the original audio clips converted to a single target speaker. val
contains clips from the same speakers as test
. dev
contains clips from different speakers than test
.
To recreate the dataset that we use in our paper:
- Download dev-clean.tar.gz and train-clean-360.tar.gz from https://www.openslr.org/12 and unzip to
llvc/LibriSpeech
python -m minimal_rvc._infer_folder \
--train_set_path "LibriSpeech/train-clean-360" \
--dev_set_path "LibriSpeech/dev-clean" \
--out_path "f_8312_ls360" \
--flatten \
--model_path "llvc_models/models/rvc/f_8312_32k-325.pth" \
--model_name "f_8312" \
--target_sr 16000 \
--f0_method "rmvpe" \
--val_percent 0.02 \
--random_seed 42 \
--f0_up_key 12
- Download test-clean.tar.gz from https://www.openslr.org/12
- Use
infer.py
to convert the test-clean folder using the checkpoint that you want to evaluate - Activate the eval environment and run
eval.py
on your converted audio and directory of ground-truth audio files.
Many of the modules written in minimal_rvc/
are based on the following repositories: