ZiyueJiang commited on
Commit
98dd72d
·
1 Parent(s): 453a650

code update for duration of ZeroGPU

Browse files
Files changed (1) hide show
  1. tts/gradio_api.py +1 -2
tts/gradio_api.py CHANGED
@@ -20,6 +20,7 @@ import gradio as gr
20
  import traceback
21
  from tts.infer_cli import MegaTTS3DiTInfer, convert_to_wav, cut_wav
22
 
 
23
  CUDA_AVAILABLE = torch.cuda.is_available()
24
  infer_pipe = MegaTTS3DiTInfer(device='cuda' if CUDA_AVAILABLE else 'cpu')
25
 
@@ -59,8 +60,6 @@ def main(inp_audio, inp_npy, inp_text, infer_timestep, p_w, t_w, processes, inpu
59
 
60
 
61
  if __name__ == '__main__':
62
- os.system('huggingface-cli download ByteDance/MegaTTS3 --local-dir ./checkpoints --repo-type model')
63
-
64
  mp.set_start_method('spawn', force=True)
65
  mp_manager = mp.Manager()
66
 
 
20
  import traceback
21
  from tts.infer_cli import MegaTTS3DiTInfer, convert_to_wav, cut_wav
22
 
23
+ os.system('huggingface-cli download ByteDance/MegaTTS3 --local-dir ./checkpoints --repo-type model')
24
  CUDA_AVAILABLE = torch.cuda.is_available()
25
  infer_pipe = MegaTTS3DiTInfer(device='cuda' if CUDA_AVAILABLE else 'cpu')
26
 
 
60
 
61
 
62
  if __name__ == '__main__':
 
 
63
  mp.set_start_method('spawn', force=True)
64
  mp_manager = mp.Manager()
65