karras samplers for img2img?

AUTOMATIC1111 · Oct 6, 2022 · b34b25b · b34b25b · Birch-san · Oct 8, 2022
1 parent 405c817
commit b34b25b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py
@@ -338,9 +338,11 @@ def sample_img2img(self, p, x, noise, conditioning, unconditional_conditioning,
         steps, t_enc = setup_img2img_steps(p, steps)
 
         if p.sampler_noise_scheduler_override:
-          sigmas = p.sampler_noise_scheduler_override(steps)
+            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)
+            sigmas = self.model_wrap.get_sigmas(steps)
 
         noise = noise * sigmas[steps - t_enc - 1]
         xi = x + noise