Commit 7ba9bc2f authored by Alex "mcmonkey" Goodwin's avatar Alex "mcmonkey" Goodwin
Browse files

fix dpm2 in img2img as well

parent 180fdf78

modules/sd_samplers.py

+3 −0
Original line number Diff line number Diff line
@@ -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]

CRA Git | Maintained and supported by SUSTech CRA and CCSE