Make style

TTS/tts/configs/speedy_speech_config.py

@@ -1,8 +1,8 @@
 from dataclasses import dataclass, field
 from typing import List
 
-from TTS.tts.models.forward_tts import ForwardTTSArgs
 from TTS.tts.configs.shared_configs import BaseTTSConfig
+from TTS.tts.models.forward_tts import ForwardTTSArgs
 
 
 @dataclass

TTS/tts/datasets/dataset.py

@@ -419,6 +419,7 @@ class TTSDataset(Dataset):
                 d_vectors = [self.d_vector_mapping[w]["embedding"] for w in wav_files_names]
             else:
                 d_vectors = None
+
             # get numerical speaker ids from speaker names
             if self.speaker_id_mapping:
                 speaker_ids = [self.speaker_id_mapping[sn] for sn in batch["speaker_name"]]

TTS/tts/models/base_tacotron.py

@@ -1,6 +1,6 @@
 import copy
 from abc import abstractmethod
-from typing import Dict, List
+from typing import Dict
 
 import torch
 from coqpit import Coqpit

TTS/tts/models/tacotron.py

@@ -258,10 +258,10 @@ class Tacotron(BaseTacotron):
                 stop_targets.float(),
                 stop_target_lengths,
                 mel_lengths,
-                outputs["decoder_outputs_backward"].float(),
+                None if outputs["decoder_outputs_backward"] is None else outputs["decoder_outputs_backward"].float(),
                 outputs["alignments"].float(),
                 alignment_lengths,
-                outputs["alignments_backward"].float(),
+                None if outputs["alignments_backward"] is None else outputs["alignments_backward"].float(),
                 text_lengths,
             )
 

TTS/tts/models/tacotron2.py

@@ -1,6 +1,7 @@
 # coding: utf-8
 
 from typing import Dict
+
 import torch
 from coqpit import Coqpit
 from torch import nn