Update inference/m4singer/base_svs_infer.py

inference/m4singer/base_svs_infer.py

@@ -40,11 +40,11 @@ class BaseSVSInfer:
         self.vocoder.eval()
         self.vocoder.to(self.device)
 
-    @spaces.GPU
+    @spaces.GPU(duration=8)
     def build_model(self):
         raise NotImplementedError
 
-    @spaces.GPU
+    @spaces.GPU(duration=8)
     def forward_model(self, inp):
         raise NotImplementedError
 
@@ -74,7 +74,7 @@ class BaseSVSInfer:
             # [T]
         return y[None]
         
-    @spaces.GPU
+    @spaces.GPU(duration=8)
     def preprocess_word_level_input(self, inp):
         # Pypinyin can't solve polyphonic words
         text_raw = inp['text']
@@ -142,7 +142,7 @@ class BaseSVSInfer:
             return None
         return ph_seq, note_lst, midi_dur_lst, is_slur
 
-    @spaces.GPU
+    @spaces.GPU(duration=8)
     def preprocess_phoneme_level_input(self, inp):
         ph_seq = inp['ph_seq']
         note_lst = inp['note_seq'].split()
@@ -157,7 +157,7 @@ class BaseSVSInfer:
             return None
         return ph_seq, note_lst, midi_dur_lst, is_slur
 
-    @spaces.GPU
+    @spaces.GPU(duration=8)
     def preprocess_input(self, inp, input_type='word'):
         """
         :param inp: {'text': str, 'item_name': (str, optional), 'spk_name': (str, optional)}
@@ -227,7 +227,7 @@ class BaseSVSInfer:
         }
         return batch
         
-    @spaces.GPU
+    @spaces.GPU(duration=8)
     def postprocess_output(self, output):
         return output