From d1173b9cdbfce7bf1cb5c30f058f16792c8739ee Mon Sep 17 00:00:00 2001 From: lym0302 Date: Tue, 14 Feb 2023 03:15:24 +0000 Subject: [PATCH] fix --- paddlespeech/t2s/models/diffsinger/diffsinger.py | 8 ++++---- paddlespeech/t2s/modules/diffusion.py | 4 +--- 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/paddlespeech/t2s/models/diffsinger/diffsinger.py b/paddlespeech/t2s/models/diffsinger/diffsinger.py index 4ca64bd94..df0549f84 100644 --- a/paddlespeech/t2s/models/diffsinger/diffsinger.py +++ b/paddlespeech/t2s/models/diffsinger/diffsinger.py @@ -270,10 +270,10 @@ class DiffSinger(nn.Layer): mel_fs2 = mel_fs2.unsqueeze(0).transpose((0, 2, 1)) cond_fs2 = self.fs2.encoder_infer(text, note, note_dur, is_slur) cond_fs2 = cond_fs2.transpose((0, 2, 1)) - mel, _ = self.diffusion(mel_fs2, cond_fs2) - # noise = paddle.randn(mel_fs2.shape) - # mel = self.diffusion.inference( - # noise=noise, cond=cond_fs2, ref_x=mel_fs2, num_inference_steps=100) + # mel, _ = self.diffusion(mel_fs2, cond_fs2) + noise = paddle.randn(mel_fs2.shape) + mel = self.diffusion.inference( + noise=noise, cond=cond_fs2, ref_x=mel_fs2, num_inference_steps=100) mel = mel.transpose((0, 2, 1)) return mel[0] diff --git a/paddlespeech/t2s/modules/diffusion.py b/paddlespeech/t2s/modules/diffusion.py index 02a412ab4..67acb6fb8 100644 --- a/paddlespeech/t2s/modules/diffusion.py +++ b/paddlespeech/t2s/modules/diffusion.py @@ -372,11 +372,9 @@ class GaussianDiffusion(nn.Layer): The noises which is added to the input. """ - # print("xxxxxxxxxxxxxxxx1: ", x, x.shape) x = x.transpose((0, 2, 1)) x = self.norm_spec(x) x = x.transpose((0, 2, 1)) - print("xxxxxxxxxxxxxxxx2: ", x, x.shape) noise_scheduler = self.noise_scheduler # Sample noise that we'll add to the mel-spectrograms @@ -411,7 +409,7 @@ class GaussianDiffusion(nn.Layer): num_inference_steps: Optional[int]=1000, strength: Optional[float]=None, scheduler_type: Optional[str]="ddpm", - clip_noise: Optional[bool]=False, + clip_noise: Optional[bool]=True, clip_noise_range: Optional[Tuple[float, float]]=(-1, 1), callback: Optional[Callable[[int, int, int, paddle.Tensor], None]]=None,