storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-28 09:08:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes to inference script to match Rust

Browse Source
This commit is contained in:
Brandon Thomas
2023-03-21 05:35:30 -04:00
parent 85d9b7afec
commit 548ce49535
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/tacotron2_v1_early_fakeyou/vocodes_inference_updated.py
+6 -6
View File
@@ -20,8 +20,8 @@ parser.add_argument('--hifigan_vocoder_checkpoint_path', type=str, help='path th
parser.add_argument('--hifigan_superres_vocoder_checkpoint_path', type=str, help='path the TTS vocoder model') parser.add_argument('--hifigan_superres_vocoder_checkpoint_path', type=str, help='path the TTS vocoder model')
# Optional mel scaling before vocoding # Optional mel scaling before vocoding
parser.add_argument('--use_default_mel_multiply_factor', type=bool, help='', action='store_true') parser.add_argument('--use_default_mel_multiply_factor', help='', action='store_true')
parser.add_argument('--maybe_custom_mel_multiply_factor', type=int, help='') parser.add_argument('--custom_mel_multiply_factor', type=float, help='')
# Premium features # Premium features
parser.add_argument('--max_decoder_steps', type=int, help='') parser.add_argument('--max_decoder_steps', type=int, help='')
@@ -47,10 +47,10 @@ import json
import numpy as np import numpy as np
import torch import torch
from hparams import create_hparams #from hparams import create_hparams
from model import Tacotron2 #from model import Tacotron2
from text import text_to_sequence #from text import text_to_sequence
from denoiser import Denoiser #from denoiser import Denoiser
from vocodes_common import FakeYouTt2Pipeline from vocodes_common import FakeYouTt2Pipeline
from vocodes_common import print_gpu_info from vocodes_common import print_gpu_info