apply to DPM2 (non-ancestral) as well

parent 8b0703b8
......@@ -494,7 +494,7 @@ class KDiffusionSampler:
x = x * sigmas[0]
if self.funcname == "sample_dpm_2_ancestral": # workaround dpm2 a issue
if self.funcname in ['sample_dpm_2_ancestral', 'sample_dpm_2']:
sigmas = torch.cat([sigmas[:-2], sigmas[-1:]])
extra_params_kwargs = self.initialize(p)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment