diff --git a/modules/processing.py b/modules/processing.py index 343eb43a..ba685349 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -1326,6 +1326,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): self.rng = rng.ImageRNG(samples.shape[1:], self.seeds, subseeds=self.subseeds, subseed_strength=self.subseed_strength, seed_resize_from_h=self.seed_resize_from_h, seed_resize_from_w=self.seed_resize_from_w) noise = self.rng.next() + samples = samples.to(noise) # GC now before running the next img2img to prevent running out of memory devices.torch_gc()