diff options
| author | Alex "mcmonkey" Goodwin <git_commits@alexgoodwin.dev> | 2022-12-18 19:16:42 -0800 |
|---|---|---|
| committer | Alex "mcmonkey" Goodwin <git_commits@alexgoodwin.dev> | 2022-12-18 19:16:42 -0800 |
| commit | 7ba9bc2fdbfae8115294962510492faafeb48573 (patch) | |
| tree | bd5621d30e034223a0b06199d7880f504bc7f97f /modules/sd_samplers.py | |
| parent | 180fdf7809ea18de2d3b04618846d5a4e33c002e (diff) | |
fix dpm2 in img2img as well
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index ae3d8bfa..1a1b8919 100644 --- 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]
|