From 0be49f49674262f4407d4d788ff19570a3bd0dd7 Mon Sep 17 00:00:00 2001 From: unknown Date: Tue, 29 Oct 2024 17:35:42 +0200 Subject: [PATCH] flip axis y in mel --- src/f5_tts/model/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/f5_tts/model/utils.py b/src/f5_tts/model/utils.py index 0314d1f..7ac4aaa 100644 --- a/src/f5_tts/model/utils.py +++ b/src/f5_tts/model/utils.py @@ -194,6 +194,7 @@ def normalize_and_colorize_spectrogram(mel_org): mel_norm = (mel_org - mel_min) / (mel_max - mel_min + 1e-8) mel_colored = plt.get_cmap("viridis")(mel_norm.detach().cpu().numpy())[:, :, :3] mel_colored = np.transpose(mel_colored, (2, 0, 1)) + mel_colored = np.flip(mel_colored, axis=1) return mel_colored