Update usr/diff/shallow_diffusion_tts.py

usr/diff/shallow_diffusion_tts.py

@@ -15,7 +15,7 @@ from modules.fastspeech.fs2 import FastSpeech2
 from modules.diffsinger_midi.fs2 import FastSpeech2MIDI
 from utils.hparams import hparams
 
-import spaces
+
 
 def exists(x):
     return x is not None
@@ -227,11 +227,9 @@ class GaussianDiffusion(nn.Module):
 
         return loss
 
-    @spaces.GPU(duration=120)
     def forward(self, txt_tokens, mel2ph=None, spk_embed=None,
                 ref_mels=None, f0=None, uv=None, energy=None, infer=False, **kwargs):
-        b, *_, device = *txt_tokens.shape, "cuda:0" #txt_tokens.device
-        print(f"在{device}上运行")
+        b, *_, device = *txt_tokens.shape, txt_tokens.device
         ret = self.fs2(txt_tokens, mel2ph, spk_embed, ref_mels, f0, uv, energy,
                        skip_decoder=(not infer), infer=infer, **kwargs)
         cond = ret['decoder_inp'].transpose(1, 2)
@@ -258,7 +256,6 @@ class GaussianDiffusion(nn.Module):
                 x = torch.randn(shape, device=device)
 
             if hparams.get('pndm_speedup'):
-                print("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',