apply to DPM2 (non-ancestral) as well

180fdf78 · Alex "mcmonkey" Goodwin · 8b0703b8 · 180fdf78
Commit 180fdf78 authored Dec 16, 2022 by Alex "mcmonkey" Goodwin
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

modules/sd_samplers.py modules/sd_samplers.py +1 -1

No files found.
--- 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)