zhzluke96 commited on
Commit
503e823
1 Parent(s): d5d0921
Files changed (1) hide show
  1. modules/webui/webui_utils.py +5 -5
modules/webui/webui_utils.py CHANGED
@@ -83,13 +83,13 @@ def segments_length_limit(
83
 
84
 
85
  @torch.inference_mode()
86
- @spaces.GPU
87
  def apply_audio_enhance(audio_data, sr, enable_denoise, enable_enhance):
88
  return _apply_audio_enhance(audio_data, sr, enable_denoise, enable_enhance)
89
 
90
 
91
  @torch.inference_mode()
92
- @spaces.GPU
93
  def synthesize_ssml(
94
  ssml: str,
95
  batch_size=4,
@@ -137,7 +137,7 @@ def synthesize_ssml(
137
 
138
 
139
  # @torch.inference_mode()
140
- @spaces.GPU
141
  def tts_generate(
142
  text,
143
  temperature=0.3,
@@ -218,14 +218,14 @@ def tts_generate(
218
 
219
 
220
  @torch.inference_mode()
221
- @spaces.GPU
222
  def refine_text(text: str, prompt: str):
223
  text = text_normalize(text)
224
  return refiner.refine_text(text, prompt=prompt)
225
 
226
 
227
  @torch.inference_mode()
228
- @spaces.GPU
229
  def split_long_text(long_text_input):
230
  spliter = SentenceSplitter(webui_config.spliter_threshold)
231
  sentences = spliter.parse(long_text_input)
 
83
 
84
 
85
  @torch.inference_mode()
86
+ @spaces.GPU(duration=120)
87
  def apply_audio_enhance(audio_data, sr, enable_denoise, enable_enhance):
88
  return _apply_audio_enhance(audio_data, sr, enable_denoise, enable_enhance)
89
 
90
 
91
  @torch.inference_mode()
92
+ @spaces.GPU(duration=120)
93
  def synthesize_ssml(
94
  ssml: str,
95
  batch_size=4,
 
137
 
138
 
139
  # @torch.inference_mode()
140
+ @spaces.GPU(duration=120)
141
  def tts_generate(
142
  text,
143
  temperature=0.3,
 
218
 
219
 
220
  @torch.inference_mode()
221
+ @spaces.GPU(duration=120)
222
  def refine_text(text: str, prompt: str):
223
  text = text_normalize(text)
224
  return refiner.refine_text(text, prompt=prompt)
225
 
226
 
227
  @torch.inference_mode()
228
+ @spaces.GPU(duration=120)
229
  def split_long_text(long_text_input):
230
  spliter = SentenceSplitter(webui_config.spliter_threshold)
231
  sentences = spliter.parse(long_text_input)