From df653f1e98cdc4516ba025a25403d08da7986824 Mon Sep 17 00:00:00 2001 From: liubaiji Date: Wed, 11 Sep 2024 10:36:32 +0800 Subject: [PATCH] [refator] modify fade_in_out func to a commom form --- cosyvoice/utils/common.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/cosyvoice/utils/common.py b/cosyvoice/utils/common.py index a5435eb..62ab916 100644 --- a/cosyvoice/utils/common.py +++ b/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)