Control diffusion-svc inference steps in args (#38)

open-mmlab · Dec 18, 2023 · 53ff15a · 53ff15a
1 parent b17eea8
commit 53ff15a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bins/svc/inference.py b/bins/svc/inference.py
@@ -205,7 +205,7 @@ def build_parser():
     parser.add_argument(
         "--diffusion_inference_steps",
         type=int,
-        default=50,
+        default=1000,
         help="Number of inference steps. Only applicable to diffusion inference.",
     )
     return parser

diff --git a/models/svc/diffusion/diffusion_inference.py b/models/svc/diffusion/diffusion_inference.py
@@ -41,7 +41,7 @@ def __init__(self, args=None, cfg=None, infer_type="from_dataset"):
         self.pipeline = DiffusionInferencePipeline(
             self.model[1],
             self.scheduler,
-            cfg.inference.diffusion.scheduler_settings.num_inference_timesteps,
+            args.diffusion_inference_steps,
         )
 
     def _build_model(self):