send noisy latent into refiner without adding noise

This commit is contained in:
AUTOMATIC1111
2023-08-07 12:10:16 +03:00
parent 3f82820612
commit 686598387f
2 changed files with 19 additions and 16 deletions

View File

@@ -276,6 +276,7 @@ class KDiffusionSampler:
self.model_wrap_cfg = CFGDenoiser(self.model_wrap)
self.sampler_noises = None
self.stop_at = None
self.noisy_output = None
self.eta = None
self.config = None # set by the function calling the constructor
self.last_latent = None
@@ -297,6 +298,7 @@ class KDiffusionSampler:
if opts.live_preview_content == "Combined":
sd_samplers_common.store_latent(latent)
self.last_latent = latent
self.noisy_output = d['x']
if self.stop_at is not None and step > self.stop_at:
raise sd_samplers_common.InterruptedException