Skip to content

l4b4r4b4b4/metavoice-src

 
 

Repository files navigation

MetaVoice-1B

Demo | Open In Colab

MetaVoice-1B is a 1.2B parameter base model trained on 100K hours of speech for TTS (text-to-speech). It has been built with the following priorities:

  • Emotional speech rhythm and tone in English.
  • Zero-shot cloning for American & British voices, with 30s reference audio.
  • Support for (cross-lingual) voice cloning with finetuning.
    • We have had success with as little as 1 minute training data for Indian speakers.
  • Support for long-form synthesis.

We’re releasing MetaVoice-1B under the Apache 2.0 license, it can be used without restrictions.

Quickstart - tl;dr

Web UI

docker-compose up -d ui && docker-compose ps && docker-compose logs -f

Server

docker-compose up -d server && docker-compose ps && docker-compose logs -f

Installation

Pre-requisites:

  • GPU VRAM >=16GB
  • Python >=3.10,<3.12

Environment setup

# install ffmpeg
wget https://johnvansickle.com/ffmpeg/builds/ffmpeg-git-amd64-static.tar.xz
wget https://johnvansickle.com/ffmpeg/builds/ffmpeg-git-amd64-static.tar.xz.md5
md5sum -c ffmpeg-git-amd64-static.tar.xz.md5
tar xvf ffmpeg-git-amd64-static.tar.xz
sudo mv ffmpeg-git-*-static/ffprobe ffmpeg-git-*-static/ffmpeg /usr/local/bin/
rm -rf ffmpeg-git-*

# install rust if not installed (ensure you've restarted your terminal after installation)
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh

pip install -r requirements.txt

# Flash Attention works only on latest Nvidia GPUs (Hopper, Ampere & Ada). If you have a different GPU (Tesla or Turing), do not install this.
pip install flash-attn

pip install -e .

Usage

  1. Download it and use it anywhere (including locally) with our reference implementation
python fam/llm/sample.py --spk_cond_path="assets/bria.mp3" --text="This is a demo of text to speech by MetaVoice-1B, an open-source foundational audio model."
  1. Deploy it on any cloud (AWS/GCP/Azure), using our inference server or web UI
python fam/llm/serving.py
python app.py 
  1. Use it via Hugging Face

Soon

  • Faster inference ⚡
  • Fine-tuning code
  • Synthesis of arbitrary length text

Architecture

We predict EnCodec tokens from text, and speaker information. This is then diffused up to the waveform level, with post-processing applied to clean up the audio.

  • We use a causal GPT to predict the first two hierarchies of EnCodec tokens. Text and audio are part of the LLM context. Speaker information is passed via conditioning at the token embedding layer. This speaker conditioning is obtained from a separately trained speaker verification network.
    • The two hierarchies are predicted in a "flattened interleaved" manner, we predict the first token of the first hierarchy, then the first token of the second hierarchy, then the second token of the first hierarchy, and so on.
    • We use condition-free sampling to boost the cloning capability of the model.
    • The text is tokenised using a custom trained BPE tokeniser with 512 tokens.
    • Note that we've skipped predicting semantic tokens as done in other works, as we found that this isn't strictly necessary.
  • We use a non-causal (encoder-style) transformer to predict the rest of the 6 hierarchies from the first two hierarchies. This is a super small model (~10Mn parameters), and has extensive zero-shot generalisation to most speakers we've tried. Since it's non-causal, we're also able to predict all the timesteps in parallel.
  • We use multi-band diffusion to generate waveforms from the EnCodec tokens. We noticed that the speech is clearer than using the original RVQ decoder or VOCOS. However, the diffusion at waveform level leaves some background artifacts which are quite unpleasant to the ear. We clean this up in the next step.
  • We use DeepFilterNet to clear up the artifacts introduced by the multi-band diffusion.

Optimizations

The model supports:

  1. KV-caching via Flash Decoding
  2. Batching (including texts of different lengths)

Contribute

Acknowledgements

We are grateful to Together.ai for their 24/7 help in marshalling our cluster. We thank the teams of AWS, GCP & Hugging Face for support with their cloud platforms.

About

AI for human-level speech intelligence

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.0%
  • Dockerfile 1.0%