update

modules/utils/hf.py

@@ -10,8 +10,11 @@ except:
         def __init__(self):
             pass
 
-        def GPU(self, fn):
-            return fn
+        def GPU(self, *args, **kwargs):
+            def _GPU(func):
+                return func
+
+            return _GPU
 
     spaces = NoneSpaces()
     is_spaces_env = False

modules/webui/speaker/speaker_creator.py

@@ -54,7 +54,7 @@ names_list = [
 
 
 @torch.inference_mode()
-@spaces.GPU
+@spaces.GPU(duration=120)
 def create_spk_from_seed(
     seed: int,
     name: str,
@@ -75,7 +75,7 @@ def create_spk_from_seed(
 
 
 @torch.inference_mode()
-@spaces.GPU
+@spaces.GPU(duration=120)
 def test_spk_voice(seed: int, text: str):
     return tts_generate(
         spk=seed,

modules/webui/speaker/speaker_editor.py

@@ -9,7 +9,7 @@ import tempfile
 
 
 @torch.inference_mode()
-@spaces.GPU
+@spaces.GPU(duration=120)
 def test_spk_voice(spk_file, text: str):
     if spk_file == "" or spk_file is None:
         return None

modules/webui/speaker/speaker_merger.py

@@ -75,7 +75,7 @@ def merge_spk(
 
 
 @torch.inference_mode()
-@spaces.GPU
+@spaces.GPU(duration=120)
 def merge_and_test_spk_voice(
     spk_a, spk_a_w, spk_b, spk_b_w, spk_c, spk_c_w, spk_d, spk_d_w, test_text
 ):
@@ -96,7 +96,7 @@ def merge_and_test_spk_voice(
 
 
 @torch.inference_mode()
-@spaces.GPU
+@spaces.GPU(duration=120)
 def merge_spk_to_file(
     spk_a,
     spk_a_w,

modules/webui/ssml/podcast_tab.py

@@ -9,7 +9,7 @@ from modules.utils.hf import spaces
 
 # NOTE: 因为 text_normalize 需要使用 tokenizer
 @torch.inference_mode()
-@spaces.GPU
+@spaces.GPU(duration=120)
 def merge_dataframe_to_ssml(msg, spk, style, df: pd.DataFrame):
     ssml = ""
     indent = " " * 2

modules/webui/ssml/spliter_tab.py

@@ -12,7 +12,7 @@ from modules.utils.hf import spaces
 
 # NOTE: 因为 text_normalize 需要使用 tokenizer
 @torch.inference_mode()
-@spaces.GPU
+@spaces.GPU(duration=120)
 def merge_dataframe_to_ssml(dataframe, spk, style, seed):
     if style == "*auto":
         style = None