apply to DPM2 (non-ancestral) as well

2022-12-16 08:42:00 -08:00 · 2022-12-16 08:42:00 -08:00 · 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)