ZiyueJiang commited on
Commit
d2c9151
·
1 Parent(s): 15868b7

update pytorch version for ZeroGPU

Browse files
Files changed (1) hide show
  1. tts/gradio_api.py +2 -2
tts/gradio_api.py CHANGED
@@ -22,7 +22,7 @@ from tts.infer_cli import MegaTTS3DiTInfer, convert_to_wav, cut_wav
22
 
23
  import spaces
24
 
25
- @spaces.GPU
26
  def model_worker(input_queue, output_queue, device_id):
27
  device = None
28
  if device_id is not None:
@@ -94,7 +94,7 @@ if __name__ == '__main__':
94
  description="Upload a speech clip as a reference for timbre, " +
95
  "upload the pre-extracted latent file, "+
96
  "input the target text, and receive the cloned voice. "+
97
- "Tip: a generation process should be within 60s (check if your input text are too long).", concurrency_limit=1)
98
  api_interface.launch(server_name='0.0.0.0', server_port=7860, debug=True)
99
  for p in processes:
100
  p.join()
 
22
 
23
  import spaces
24
 
25
+ @spaces.GPU(duration=120)
26
  def model_worker(input_queue, output_queue, device_id):
27
  device = None
28
  if device_id is not None:
 
94
  description="Upload a speech clip as a reference for timbre, " +
95
  "upload the pre-extracted latent file, "+
96
  "input the target text, and receive the cloned voice. "+
97
+ "Tip: a generation process should be within 120s (check if your input text are too long).", concurrency_limit=1)
98
  api_interface.launch(server_name='0.0.0.0', server_port=7860, debug=True)
99
  for p in processes:
100
  p.join()