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

add --f0-method to inference entrypoint

Browse Source
This commit is contained in:
Brandon Thomas
2023-05-18 23:05:14 -07:00
parent ab8454c44d
commit 7f60d4ca25
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
voice_conversion/so-vits-svc-paul-v2/src/fakeyou_infer.py
+9
View File
@@ -165,6 +165,13 @@ def cli():
"-r", "--cluster-infer-ratio", type=float, default=0, help="cluster infer ratio" "-r", "--cluster-infer-ratio", type=float, default=0, help="cluster infer ratio"
) )
@click.option("-n", "--noise-scale", type=float, default=0.4, help="noise scale") @click.option("-n", "--noise-scale", type=float, default=0.4, help="noise scale")
@click.option(
"-fm",
"--f0-method",
type=click.Choice(["crepe", "crepe-tiny", "parselmouth", "dio", "harvest"]),
default="dio",
help="f0 prediction method",
)
@click.option("-p", "--pad-seconds", type=float, default=0.5, help="pad seconds") @click.option("-p", "--pad-seconds", type=float, default=0.5, help="pad seconds")
@click.option( @click.option(
"-d", "-d",
@@ -189,6 +196,7 @@ def infer(
auto_predict_f0: bool = False, auto_predict_f0: bool = False,
cluster_infer_ratio: float = 0, cluster_infer_ratio: float = 0,
noise_scale: float = 0.4, noise_scale: float = 0.4,
f0_method: Literal["crepe", "crepe-tiny", "parselmouth", "dio", "harvest"] = "dio",
pad_seconds: float = 0.5, pad_seconds: float = 0.5,
chunk_seconds: float = 0.5, chunk_seconds: float = 0.5,
absolute_thresh: bool = False, absolute_thresh: bool = False,
@@ -234,6 +242,7 @@ def infer(
auto_predict_f0=auto_predict_f0, auto_predict_f0=auto_predict_f0,
cluster_infer_ratio=cluster_infer_ratio, cluster_infer_ratio=cluster_infer_ratio,
noise_scale=noise_scale, noise_scale=noise_scale,
f0_method=f0_method,
pad_seconds=pad_seconds, pad_seconds=pad_seconds,
chunk_seconds=chunk_seconds, chunk_seconds=chunk_seconds,
absolute_thresh=absolute_thresh, absolute_thresh=absolute_thresh,