Update inference/svs/ds_e2e.py

inference/svs/ds_e2e.py

@@ -9,6 +9,7 @@ from usr.diffsinger_task import DIFF_DECODERS
 from modules.fastspeech.pe import PitchExtractor
 import utils
 
+import spaces
 
 class DiffSingerE2EInfer(BaseSVSInfer):
     def build_model(self):
@@ -29,6 +30,7 @@ class DiffSingerE2EInfer(BaseSVSInfer):
             self.pe.eval()
         return model
 
+    @spaces.GPU(duration=180)
     def forward_model(self, inp):
         sample = self.input_to_batch(inp)
         txt_tokens = sample['txt_tokens']  # [B, T_t]