Prechádzať zdrojové kódy

[refator] modify fade_in_out func to a commom form

liubaiji 1 rok pred
rodič
commit
df653f1e98
1 zmenil súbory, kde vykonal 3 pridanie a 2 odobranie
  1. 3 2
      cosyvoice/utils/common.py

+ 3 - 2
cosyvoice/utils/common.py

@@ -139,6 +139,7 @@ def fade_in_out(fade_in_mel, fade_out_mel, window):
     device = fade_in_mel.device
     fade_in_mel, fade_out_mel = fade_in_mel.cpu(), fade_out_mel.cpu()
     mel_overlap_len = int(window.shape[0] / 2)
-    fade_in_mel[:, :, :mel_overlap_len] = fade_in_mel[:, :, :mel_overlap_len] * window[:mel_overlap_len] + \
-        fade_out_mel[:, :, -mel_overlap_len:] * window[mel_overlap_len:]
+
+    fade_in_mel[..., :mel_overlap_len] = fade_in_mel[..., :mel_overlap_len] * window[:mel_overlap_len] + \
+        fade_out_mel[..., -mel_overlap_len:] * window[mel_overlap_len:]
     return fade_in_mel.to(device)