SWivid commited on
Commit
5b10099
·
2 Parent(s): 2ccfbdc 0be49f4

Merge branch 'main' of github.com:lpscr/F5-TTS into lpscr-main

Browse files
Files changed (1) hide show
  1. src/f5_tts/model/utils.py +1 -0
src/f5_tts/model/utils.py CHANGED
@@ -194,6 +194,7 @@ def normalize_and_colorize_spectrogram(mel_org):
194
  mel_norm = (mel_org - mel_min) / (mel_max - mel_min + 1e-8)
195
  mel_colored = plt.get_cmap("viridis")(mel_norm.detach().cpu().numpy())[:, :, :3]
196
  mel_colored = np.transpose(mel_colored, (2, 0, 1))
 
197
  return mel_colored
198
 
199
 
 
194
  mel_norm = (mel_org - mel_min) / (mel_max - mel_min + 1e-8)
195
  mel_colored = plt.get_cmap("viridis")(mel_norm.detach().cpu().numpy())[:, :, :3]
196
  mel_colored = np.transpose(mel_colored, (2, 0, 1))
197
+ mel_colored = np.flip(mel_colored, axis=1)
198
  return mel_colored
199
 
200