Update usr/diff/shallow_diffusion_tts.py

usr/diff/shallow_diffusion_tts.py

@@ -226,8 +226,8 @@ class GaussianDiffusion(nn.Module):
             raise NotImplementedError()
 
         return loss
-        
-    @spaces.GPU(duration=180)
+
+    @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, txt_tokens.device
@@ -318,4 +318,4 @@ class OfflineGaussianDiffusion(GaussianDiffusion):
                 x = self.p_sample(x, torch.full((b,), i, device=device, dtype=torch.long), cond)
             x = x[:, 0].transpose(1, 2)
             ret['mel_out'] = self.denorm_spec(x)
-        return ret
+        return ret