Commit b34b25b4 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

karras samplers for img2img?

parent 405c8171

modules/sd_samplers.py

+4 −2
Original line number Diff line number Diff line
@@ -339,6 +339,8 @@ class KDiffusionSampler: 


        if p.sampler_noise_scheduler_override:

            sigmas = p.sampler_noise_scheduler_override(steps)

        elif self.config is not None and self.config.options.get('scheduler', None) == 'karras':

            sigmas = k_diffusion.sampling.get_sigmas_karras(n=steps, sigma_min=0.1, sigma_max=10, device=shared.device)

        else:

            sigmas = self.model_wrap.get_sigmas(steps)

CRA Git | Maintained and supported by SUSTech CRA and CCSE