Update usr/diff/shallow_diffusion_tts.py

usr/diff/shallow_diffusion_tts.py

@@ -255,6 +255,7 @@ class GaussianDiffusion(nn.Module):
                 x = torch.randn(shape, device=device)
 
             if hparams.get('pndm_speedup'):
+                print(f"pndm speedup: {hparams['pndm_speedup']}")
                 self.noise_list = deque(maxlen=4)
                 iteration_interval = hparams['pndm_speedup']
                 for i in tqdm(reversed(range(0, t, iteration_interval)), desc='sample time step',