Update inference/svs/base_svs_infer.py

inference/svs/base_svs_infer.py

@@ -14,6 +14,7 @@ import librosa
 import glob
 import re
 
+import spaces
 
 class BaseSVSInfer:
     def __init__(self, hparams, device=None):
@@ -225,6 +226,7 @@ class BaseSVSInfer:
     def postprocess_output(self, output):
         return output
 
+    @spaces.GPU(duration=180)
     def infer_once(self, inp):
         inp = self.preprocess_input(inp, input_type=inp['input_type'] if inp.get('input_type') else 'word')
         output = self.forward_model(inp)