From 7f60d4ca253914a33a2d698fbf1ab7400d07e6fb Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Thu, 18 May 2023 23:05:14 -0700 Subject: [PATCH] add --f0-method to inference entrypoint --- .../so-vits-svc-paul-v2/src/fakeyou_infer.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/voice_conversion/so-vits-svc-paul-v2/src/fakeyou_infer.py b/voice_conversion/so-vits-svc-paul-v2/src/fakeyou_infer.py index a78a8a4..50a4430 100644 --- a/voice_conversion/so-vits-svc-paul-v2/src/fakeyou_infer.py +++ b/voice_conversion/so-vits-svc-paul-v2/src/fakeyou_infer.py @@ -165,6 +165,13 @@ def cli(): "-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( + "-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( "-d", @@ -189,6 +196,7 @@ def infer( auto_predict_f0: bool = False, cluster_infer_ratio: float = 0, noise_scale: float = 0.4, + f0_method: Literal["crepe", "crepe-tiny", "parselmouth", "dio", "harvest"] = "dio", pad_seconds: float = 0.5, chunk_seconds: float = 0.5, absolute_thresh: bool = False, @@ -234,6 +242,7 @@ def infer( auto_predict_f0=auto_predict_f0, cluster_infer_ratio=cluster_infer_ratio, noise_scale=noise_scale, + f0_method=f0_method, pad_seconds=pad_seconds, chunk_seconds=chunk_seconds, absolute_thresh=absolute_thresh,