karras samplers for img2img?

3 years ago · b34b25b4c9
parent 405c8171d1
commit b34b25b4c9
1 changed files with 4 additions and 2 deletions
--- a/modules/sd_samplers.py
+++ b/modules/sd_samplers.py
@ -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)