auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-04-23 08:14:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small clean up

Browse Source
This commit is contained in:
Michael Hansen
2023-02-14 10:13:18 -06:00
parent acc3068176
commit aab406cfdd
4 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/python/larynx_train/infer_generator.py
View File

@@ -8,7 +8,6 @@ from pathlib import Path
import torch
from .vits.lightning import VitsModel
from .vits.utils import audio_float_to_int16
from .vits.wavfile import write as write_wav
@@ -19,9 +18,7 @@ def main():
"""Main entry point"""
logging.basicConfig(level=logging.DEBUG)
parser = argparse.ArgumentParser(prog="larynx_train.infer_generator")
parser.add_argument(
"--model", required=True, help="Path to generator (.pt)"
)
parser.add_argument("--model", required=True, help="Path to generator (.pt)")
parser.add_argument("--output-dir", required=True, help="Path to write WAV files")
parser.add_argument("--sample-rate", type=int, default=22050)
args = parser.parse_args()