Update ctm_inference.py

sony · Aug 29, 2024 · 6ccd44d · 6ccd44d
1 parent c3edf59
commit 6ccd44d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()
+    main()