tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Format code

Browse Source
This commit is contained in:
github-actions[bot]
2023-09-09 03:35:56 +00:00
parent 40c2fae6dc
commit 33944b7a45
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
webui.py
View File

@@ -25,15 +25,13 @@ import webbrowser
net_g = None net_g = None
if ( if sys.platform == "darwin" and torch.backends.mps.is_available():
sys.platform == "darwin"
and torch.backends.mps.is_available()
):
device = "mps" device = "mps"
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1" os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
else: else:
device = "cuda" device = "cuda"
def get_text(text, language_str, hps): def get_text(text, language_str, hps):
norm_text, phone, tone, word2ph = clean_text(text, language_str) norm_text, phone, tone, word2ph = clean_text(text, language_str)
phone, tone, language = cleaned_text_to_sequence(phone, tone, language_str) phone, tone, language = cleaned_text_to_sequence(phone, tone, language_str)
@@ -207,7 +205,15 @@ if __name__ == "__main__":
btn.click( btn.click(
tts_fn, tts_fn,
inputs=[text, speaker, sdp_ratio, noise_scale, noise_scale_w, length_scale, language], inputs=[
text,
speaker,
sdp_ratio,
noise_scale,
noise_scale_w,
length_scale,
language,
],
outputs=[text_output, audio_output], outputs=[text_output, audio_output],
) )