ZiyueJiang commited on
Commit
e40c637
·
1 Parent(s): 77889cd

update head of readme

Browse files
Files changed (1) hide show
  1. tts/gradio_api.py +2 -3
tts/gradio_api.py CHANGED
@@ -20,9 +20,6 @@ import gradio as gr
20
  import traceback
21
  from tts.infer_cli import MegaTTS3DiTInfer, convert_to_wav, cut_wav
22
 
23
- from huggingface_hub import hf_hub_download
24
- hf_hub_download(repo_id="ByteDance/MegaTTS3", allow_patterns='./*', local_dir="./checkpoints", local_dir_use_symlinks="auto")
25
-
26
 
27
  def model_worker(input_queue, output_queue, device_id):
28
  device = None
@@ -61,6 +58,8 @@ def main(inp_audio, inp_npy, inp_text, infer_timestep, p_w, t_w, processes, inpu
61
 
62
 
63
  if __name__ == '__main__':
 
 
64
  mp.set_start_method('spawn', force=True)
65
  devices = os.environ.get('CUDA_VISIBLE_DEVICES', '')
66
  if devices != '':
 
20
  import traceback
21
  from tts.infer_cli import MegaTTS3DiTInfer, convert_to_wav, cut_wav
22
 
 
 
 
23
 
24
  def model_worker(input_queue, output_queue, device_id):
25
  device = None
 
58
 
59
 
60
  if __name__ == '__main__':
61
+ os.system('huggingface-cli download ByteDance/MegaTTS3 --local-dir ./checkpoints --repo-type model')
62
+
63
  mp.set_start_method('spawn', force=True)
64
  devices = os.environ.get('CUDA_VISIBLE_DEVICES', '')
65
  if devices != '':