From 6ccd44d5306bb79ba5c53240fc885715c25b7d66 Mon Sep 17 00:00:00 2001 From: Koichi Saito <116609740+koichi-saito-sony@users.noreply.github.com> Date: Thu, 29 Aug 2024 10:47:03 -0400 Subject: [PATCH] Update ctm_inference.py --- ctm_inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ctm_inference.py b/ctm_inference.py index 39ca4e6..12774c8 100644 --- a/ctm_inference.py +++ b/ctm_inference.py @@ -70,7 +70,7 @@ def parse_args(): ) parser.add_argument( "--sampler", type=str, default='determinisitc', - help="Inference sampling methods. You can choose ['determinisitc' (gamma=0), 'cm_multistep' (gamma=1), 'gamma_multistep']." + help="Inference sampling methods. You can choose ['determinisitc' (gamma=0), 'cm_multistep' (gamma=1), 'gamma_multistep', 'onestep']." ) parser.add_argument( "--sampling_gamma", type=float, default=0.9, @@ -212,4 +212,4 @@ def main(): sf.write("{}/{}.wav".format(output_dir, filename), wav, samplerate=args.sampling_rate) if __name__ == "__main__": - main() \ No newline at end of file + main()