diff --git a/api.py b/api.py index 41ddbf660..8a476cd6d 100644 --- a/api.py +++ b/api.py @@ -19,16 +19,17 @@ from text.cleaner import clean_text from module.mel_processing import spectrogram_torch from my_utils import load_audio +from config import python_exec, infer_device, is_half, api_port -DEFAULT_PORT = 9880 +DEFAULT_PORT = api_port DEFAULT_CNHUBERT = "GPT_SoVITS/pretrained_models/chinese-hubert-base" DEFAULT_BERT = "GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large" -DEFAULT_HALF = True +DEFAULT_HALF = is_half DEFAULT_GPT = "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt" DEFAULT_SOVITS = "GPT_SoVITS/pretrained_models/s2G488k.pth" -AVAILABLE_COMPUTE = "cuda" if torch.cuda.is_available() else "cpu" +# AVAILABLE_COMPUTE = "cuda" if torch.cuda.is_available() else "cpu" parser = argparse.ArgumentParser(description="GPT-SoVITS api") @@ -40,7 +41,7 @@ parser.add_argument("-dt", "--default_refer_text", type=str, default="", help="默认参考音频文本") parser.add_argument("-dl", "--default_refer_language", type=str, default="", help="默认参考音频语种") -parser.add_argument("-d", "--device", type=str, default=AVAILABLE_COMPUTE, help="cuda / cpu") +parser.add_argument("-d", "--device", type=str, default=infer_device, help="cuda / cpu") parser.add_argument("-p", "--port", type=int, default=DEFAULT_PORT, help="default: 9880") parser.add_argument("-a", "--bind_addr", type=str, default="127.0.0.1", help="default: 127.0.0.1") parser.add_argument("-hp", "--half_precision", action='store_true', default=False) @@ -253,14 +254,9 @@ def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language) yield hps.data.sampling_rate, (np.concatenate(audio_opt, 0) * 32768).astype(np.int16) -def restart(): - python = sys.executable - os.execl(python, python, *sys.argv) - - def handle(command, refer_wav_path, prompt_text, prompt_language, text, text_language): if command == "/restart": - restart() + os.execl(python_exec, python_exec, *sys.argv) elif command == "/exit": os.kill(os.getpid(), signal.SIGTERM) exit(0)