apply to DPM2 (non-ancestral) as well

3 years ago · 180fdf7809
parent 8b0703b8fc
commit 180fdf7809
1 changed files with 1 additions and 1 deletions
--- a/modules/sd_samplers.py
+++ b/modules/sd_samplers.py
@ -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)