fix dpm2 in img2img as well

2022-12-18 19:16:42 -08:00 · 2022-12-18 19:16:42 -08:00 · 7ba9bc2fdb
commit 7ba9bc2fdb
parent 180fdf7809
1 changed files with 3 additions and 0 deletions
--- a/modules/sd_samplers.py
+++ b/modules/sd_samplers.py
@ -454,6 +454,9 @@ class KDiffusionSampler:
        else:
            sigmas = self.model_wrap.get_sigmas(steps)

+        if self.funcname in ['sample_dpm_2_ancestral', 'sample_dpm_2']:
+            sigmas = torch.cat([sigmas[:-2], sigmas[-1:]])
+
        sigma_sched = sigmas[steps - t_enc - 1:]
        xi = x + noise * sigma_sched[0]