2.p

miraymen/pix2surf2.pJun 10, 2024by Flux9665

AI Summary

This release introduces prompting capabilities to control the emotional tone of synthesized speech, allowing users to condition models on emotional prompts during training and inference. It provides the necessary tools and training interfaces to fine-tune models for expressive voice generation.

Key Highlights

  • Emotional prompting control for training and inference
  • Sentence embedding extractor integration
  • Specific training pipeline support (ToucanTTS_Sent_Finetuning)
  • Demo spaces available for interactive testing
  • Utilities for extracting emotion and speaker embeddings

New Features

  • Emotional TTS control via text prompts
  • Sentence embedding extraction utilities
  • Fine-tuning interfaces for emotional speech
  • Demo environments for testing emotional transfer

Full Release Notes

In this release you can condition your TTS model on emotional prompts during training and transfer the emotion in any prompt to synthesized speech during inference.

Demo samples are available at https://anondemos.github.io/Prompting/
A demo space is available at https://huggingface.co/spaces/Thommy96/promptingtoucan

Using pretrained models:
You can use the pretrained models for inference by simply providing an instance of the sentence embedding extractor, a speaker id and a prompt (see run_sent_emb_test_suite.py).

Training your own model:
You will need to extract a number of prompts and their sentence embeddings for all emotion categories which you want to include during training (see e.g. extract_yelp_sent_embs.py).
Then in your training pipeline you need to load these sentence embeddings and pass them to the train loop. You should also provide the dimensionality of the embeddings in the instantiation of the TTS model and set static_speaker_embedding=True (see TrainingInterfaces\TrainingPipelines\ToucanTTS_Sent_Finetuning.py). Depending on how many speakers there are in the datasets you use for training, you need to adapt the dimensionality of the speaker embedding table in the TTS model. Finally you should check if the datasets you use are included in the functions for extracting emotion and speaker id from the filepath (Utility\utils.py).