2.p
nari-labs/dia22.pJun 10, 2024by Flux9665
AI Summary
This release introduces prompt-based emotional conditioning for TTS models, allowing users to synthesize speech with specific emotions during both training and inference.
Key Highlights
- Emotional prompt conditioning during training
- Transfer of emotion from prompts to synthesized speech during inference
- New demo samples and interactive spaces available
- Integration with sentence embedding extractors
New Features
- Prompting Controlled Emotional TTS
- Sentence embedding extractor support
- Training pipeline for emotion-specific models
- Demo samples and spaces
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).