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

update head of readme

Browse files
Files changed (2) hide show
  1. README.md +2 -0
  2. tts/gradio_api.py +1 -2
README.md CHANGED
@@ -8,6 +8,8 @@ sdk: "gradio"
8
  sdk_version: "5.23.3"
9
  app_file: ./tts/gradio_api.py
10
  pinned: true
 
 
11
  ---
12
 
13
 
 
8
  sdk_version: "5.23.3"
9
  app_file: ./tts/gradio_api.py
10
  pinned: true
11
+ preload_from_hub:
12
+ - ByteDance/MegaTTS3
13
  ---
14
 
15
 
tts/gradio_api.py CHANGED
@@ -21,8 +21,7 @@ import traceback
21
  from tts.infer_cli import MegaTTS3DiTInfer, convert_to_wav, cut_wav
22
 
23
  from huggingface_hub import hf_hub_download
24
-
25
- hf_hub_download(repo_id="ByteDance/MegaTTS3", local_dir="./checkpoints")
26
 
27
 
28
  def model_worker(input_queue, output_queue, device_id):
 
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):