diff --git a/modules/models/diffusion/uni_pc/sampler.py b/modules/models/diffusion/uni_pc/sampler.py index 6dd7c7fd8..953e786db 100644 --- a/modules/models/diffusion/uni_pc/sampler.py +++ b/modules/models/diffusion/uni_pc/sampler.py @@ -47,7 +47,7 @@ class UniPCSampler(object): # actual number of steps we'll run self.steps = max( - num_inference_steps - init_timestep, + init_timestep, shared.opts.uni_pc_order+1, )