Commit
·
e40c637
1
Parent(s):
77889cd
update head of readme
Browse files- 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 != '':
|