# Fine-tuning SpeechT5 Now that you are familiar with the text-to-speech task and internal workings of the SpeechT5 model that was pre-trained on English language data, let's see how we can fine-tune it to another language. ## House-keeping Make sure that you have a GPU if you want to reproduce this example. In a notebook, you can check with the following command: ```bash nvidia-smi ``` In our example we will be using approximately 40 hours of training data. If you'd like to follow along using the Google Colab free tier GPU, you will need to reduce the amount of training data to approximately 10-15 hours, and reduce the number of training steps. You'll also need some additional dependencies: ```bash pip install transformers datasets soundfile speechbrain accelerate ``` Finally, don't forget to log in to your Hugging Face account so that you could upload and share your model with the community: ```py from huggingface_hub import notebook_login notebook_login() ``` ## The dataset For this example we'll take the Dutch (`nl`) language subset of the [VoxPopuli](https://huggingface.co/datasets/qmeeus/voxpopuli) dataset. [VoxPopuli](https://huggingface.co/datasets/qmeeus/voxpopuli) is a large-scale multilingual speech corpus consisting of data sourced from 2009-2020 European Parliament event recordings. It contains labelled audio-transcription data for 15 European languages. While we will be using the Dutch language subset, feel free to pick another subset. This is an automated speech recognition (ASR) dataset, so, as mentioned before, it is not the most suitable option for training TTS models. However, it will be good enough for this exercise. Let's load the data: ```python from datasets import load_dataset, Audio dataset = load_dataset("qmeeus/voxpopuli", "nl", split="train") len(dataset) ``` **Output:** ```out 20968 ``` 20968 examples should be sufficient for fine-tuning. SpeechT5 expects audio data to have a sampling rate of 16 kHz, so make sure the examples in the dataset meet this requirement: ```python dataset = dataset.cast_column("audio", Audio(sampling_rate=16000)) ``` ## Preprocessing the data Let's begin by defining the model checkpoint to use and loading the appropriate processor that contains both tokenizer, and feature extractor that we will need to prepare the data for training: ```py from transformers import SpeechT5Processor checkpoint = "microsoft/speecht5_tts" processor = SpeechT5Processor.from_pretrained(checkpoint) ``` ### Text cleanup for SpeechT5 tokenization First, for preparing the text, we'll need the tokenizer part of the processor, so let's get it: ```py tokenizer = processor.tokenizer ``` Let's take a look at an example: ```python dataset[0] ``` **Output:** ```out {'audio_id': '20100210-0900-PLENARY-3-nl_20100210-09:06:43_4', 'language': 9, 'audio': {'path': '/root/.cache/huggingface/datasets/downloads/extracted/02ec6a19d5b97c03e1379250378454dbf3fa2972943504a91c7da5045aa26a89/train_part_0/20100210-0900-PLENARY-3-nl_20100210-09:06:43_4.wav', 'array': array([ 4.27246094e-04, 1.31225586e-03, 1.03759766e-03, ..., -9.15527344e-05, 7.62939453e-04, -2.44140625e-04]), 'sampling_rate': 16000}, 'text': 'dat kan naar mijn gevoel alleen met een brede meerderheid die wij samen zoeken.', 'gender': 'female', 'speaker_id': '1122', 'is_gold_transcript': True, 'accent': 'None'} ``` The dataset examples contain a `text` feature which we'll use as the text input. It's important to know that the SpeechT5 tokenizer doesn't have any tokens for numbers, so if your text contains numbers, you may need to write them out as text. Because SpeechT5 was trained on the English language, it may not recognize certain characters in the Dutch dataset. If left as is, these characters will be converted to `` tokens. However, in Dutch, certain characters like `à` are used to stress syllables. In order to preserve the meaning of the text, we can replace this character with a regular `a`. To identify unsupported tokens, extract all unique characters in the dataset using the `SpeechT5Tokenizer` which works with characters as tokens. To do this, we'll write the `extract_all_chars` mapping function that concatenates the transcriptions from all examples into one string and converts it to a set of characters. Make sure to set `batched=True` and `batch_size=-1` in `dataset.map()` so that all transcriptions are available at once for the mapping function. ```py def extract_all_chars(batch): all_text = " ".join(batch["text"]) vocab = list(set(all_text)) return {"vocab": [vocab], "all_text": [all_text]} vocabs = dataset.map( extract_all_chars, batched=True, batch_size=-1, keep_in_memory=True, remove_columns=dataset.column_names, ) dataset_vocab = set(vocabs["vocab"][0]) tokenizer_vocab = {k for k, _ in tokenizer.get_vocab().items()} ``` Now you have two sets of characters: one with the vocabulary from the dataset and one with the vocabulary from the tokenizer. To identify any unsupported characters in the dataset, you can take the difference between these two sets. The resulting set will contain the characters that are in the dataset but not in the tokenizer. ```py dataset_vocab - tokenizer_vocab ``` **Output:** ```out {' ', 'à', 'ç', 'è', 'ë', 'í', 'ï', 'ö', 'ü'} ``` To handle the unsupported characters identified in the previous step, we can define a function that maps these characters to valid tokens. Note that spaces are already replaced by `▁` in the tokenizer and don't need to be handled separately. ```py replacements = [ ("à", "a"), ("ç", "c"), ("è", "e"), ("ë", "e"), ("í", "i"), ("ï", "i"), ("ö", "o"), ("ü", "u"), ] def cleanup_text(inputs): for src, dst in replacements: inputs["text"] = inputs["text"].replace(src, dst) return inputs dataset = dataset.map(cleanup_text) ``` Now that we have dealt with special characters in the text, it's time to shift the focus to the audio data. ### Speakers The VoxPopuli dataset includes speech from multiple speakers, but how many speakers are represented in the dataset? To determine this, we can count the number of unique speakers and the number of examples each speaker contributes to the dataset. With a total of 20,968 examples in the dataset, this information will give us a better understanding of the distribution of speakers and examples in the data. ```py from collections import defaultdict speaker_counts = defaultdict(int) for speaker_id in dataset["speaker_id"]: speaker_counts[speaker_id] += 1 ``` By plotting a histogram you can get a sense of how much data there is for each speaker. ```py import matplotlib.pyplot as plt plt.figure() plt.hist(speaker_counts.values(), bins=20) plt.ylabel("Speakers") plt.xlabel("Examples") plt.show() ``` The histogram reveals that approximately one-third of the speakers in the dataset have fewer than 100 examples, while around ten speakers have more than 500 examples. To improve training efficiency and balance the dataset, we can limit the data to speakers with between 100 and 400 examples. ```py def select_speaker(speaker_id): return 100 Side note: If you find this spectrogram confusing, it may be due to your familiarity with the convention of placing low frequencies at the bottom and high frequencies at the top of a plot. However, when plotting spectrograms as an image using the matplotlib library, the y-axis is flipped and the spectrograms appear upside down. Now we need to apply the processing function to the entire dataset. This will take between 5 and 10 minutes. ```py dataset = dataset.map(prepare_dataset, remove_columns=dataset.column_names) ``` You'll see a warning saying that some examples in the dataset are longer than the maximum input length the model can handle (600 tokens). Remove those examples from the dataset. Here we go even further and to allow for larger batch sizes we remove anything over 200 tokens. ```py def is_not_too_long(input_ids): input_length = len(input_ids) return input_length Dict[str, torch.Tensor]: input_ids = [{"input_ids": feature["input_ids"]} for feature in features] label_features = [{"input_values": feature["labels"]} for feature in features] speaker_features = [feature["speaker_embeddings"] for feature in features] # collate the inputs and targets into a batch batch = processor.pad( input_ids=input_ids, labels=label_features, return_tensors="pt" ) # replace padding with -100 to ignore loss correctly batch["labels"] = batch["labels"].masked_fill( batch.decoder_attention_mask.unsqueeze(-1).ne(1), -100 ) # not used during fine-tuning del batch["decoder_attention_mask"] # round down target lengths to multiple of reduction factor if model.config.reduction_factor > 1: target_lengths = torch.tensor( [len(feature["input_values"]) for feature in label_features] ) target_lengths = target_lengths.new( [ length - length % model.config.reduction_factor for length in target_lengths ] ) max_length = max(target_lengths) batch["labels"] = batch["labels"][:, :max_length] # also add in the speaker embeddings batch["speaker_embeddings"] = torch.tensor(speaker_features) return batch ``` In SpeechT5, the input to the decoder part of the model is reduced by a factor 2. In other words, it throws away every other timestep from the target sequence. The decoder then predicts a sequence that is twice as long. Since the original target sequence length may be odd, the data collator makes sure to round the maximum length of the batch down to be a multiple of 2. ```py data_collator = TTSDataCollatorWithPadding(processor=processor) ``` ## Train the model Load the pre-trained model from the same checkpoint as you used for loading the processor: ```py from transformers import SpeechT5ForTextToSpeech model = SpeechT5ForTextToSpeech.from_pretrained(checkpoint) ``` The `use_cache=True` option is incompatible with gradient checkpointing. Disable it for training, and re-enable cache for generation to speed-up inference time: ```py from functools import partial # disable cache during training since it's incompatible with gradient checkpointing model.config.use_cache = False # set language and task for generation and re-enable cache model.generate = partial(model.generate, use_cache=True) ``` Define the training arguments. Here we are not computing any evaluation metrics during the training process, we'll talk about evaluation later in this chapter. Instead, we'll only look at the loss: ```python from transformers import Seq2SeqTrainingArguments training_args = Seq2SeqTrainingArguments( output_dir="speecht5_finetuned_voxpopuli_nl", # change to a repo name of your choice per_device_train_batch_size=4, gradient_accumulation_steps=8, learning_rate=1e-5, warmup_steps=500, max_steps=4000, gradient_checkpointing=True, fp16=True, eval_strategy="steps", per_device_eval_batch_size=2, save_steps=1000, eval_steps=1000, logging_steps=25, report_to=["tensorboard"], load_best_model_at_end=True, greater_is_better=False, label_names=["labels"], push_to_hub=True, ) ``` Instantiate the `Trainer` object and pass the model, dataset, and data collator to it. ```py from transformers import Seq2SeqTrainer trainer = Seq2SeqTrainer( args=training_args, model=model, train_dataset=dataset["train"], eval_dataset=dataset["test"], data_collator=data_collator, tokenizer=processor, ) ``` And with that, we're ready to start training! Training will take several hours. Depending on your GPU, it is possible that you will encounter a CUDA "out-of-memory" error when you start training. In this case, you can reduce the `per_device_train_batch_size` incrementally by factors of 2 and increase `gradient_accumulation_steps` by 2x to compensate. ```py trainer.train() ``` Push the final model to the 🤗 Hub: ```py trainer.push_to_hub() ``` ## Inference Once you have fine-tuned a model, you can use it for inference! Load the model from the 🤗 Hub (make sure to use your account name in the following code snippet): ```py model = SpeechT5ForTextToSpeech.from_pretrained( "YOUR_ACCOUNT/speecht5_finetuned_voxpopuli_nl" ) ``` Pick an example, here we'll take one from the test dataset. Obtain a speaker embedding. ```py example = dataset["test"][304] speaker_embeddings = torch.tensor(example["speaker_embeddings"]).unsqueeze(0) ``` Define some input text and tokenize it. ```py text = "hallo allemaal, ik praat nederlands. groetjes aan iedereen!" ``` Preprocess the input text: ```py inputs = processor(text=text, return_tensors="pt") ``` Instantiate a vocoder and generate speech: ```py from transformers import SpeechT5HifiGan vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan") speech = model.generate_speech(inputs["input_ids"], speaker_embeddings, vocoder=vocoder) ``` Ready to listen to the result? ```py from IPython.display import Audio Audio(speech.numpy(), rate=16000) ``` Obtaining satisfactory results from this model on a new language can be challenging. The quality of the speaker embeddings can be a significant factor. Since SpeechT5 was pre-trained with English x-vectors, it performs best when using English speaker embeddings. If the synthesized speech sounds poor, try using a different speaker embedding. Increasing the training duration is also likely to enhance the quality of the results. Even so, the speech clearly is Dutch instead of English, and it does capture the voice characteristics of the speaker (compare to the original audio in the example). Another thing to experiment with is the model's configuration. For example, try using `config.reduction_factor = 1` to see if this improves the results. In the next section, we'll talk about how we evaluate text-to-speech models.