mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

apply to DPM2 (non-ancestral) as well

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin 2022-12-16 08:42:00 -08:00
parent 8b0703b8fc
commit 180fdf7809
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers.py
View File

@ -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)