black formatted

k2-fsa · Dec 6, 2024 · ce73643 · ce73643
1 parent 60c5a1d
commit ce73643
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/egs/wenetspeech4tts/TTS/valle/valle.py b/egs/wenetspeech4tts/TTS/valle/valle.py
@@ -1670,9 +1670,7 @@ def visualize(
         text_tokens = batch["text_tokens"].to("cpu").detach().numpy()
         text_tokens_lens = batch["text_tokens_lens"].to("cpu").detach().numpy()
         audio_features = batch["audio_features"].to("cpu").detach().numpy()
-        audio_features_lens = (
-            batch["audio_features_lens"].to("cpu").detach().numpy()
-        )
+        audio_features_lens = batch["audio_features_lens"].to("cpu").detach().numpy()
         assert text_tokens.ndim == 2
 
         utt_ids, texts = batch["utt_id"], batch["text"]
@@ -1681,9 +1679,7 @@ def visualize(
         decoder_outputs = predicts[1]
         if isinstance(decoder_outputs, list):
             decoder_outputs = decoder_outputs[-1]
-        decoder_outputs = (
-            decoder_outputs.to("cpu").type(torch.float32).detach().numpy()
-        )
+        decoder_outputs = decoder_outputs.to("cpu").type(torch.float32).detach().numpy()
 
         vmin, vmax = 0, 1024  # Encodec
         if decoder_outputs.dtype == np.float32: