This commit is contained in:
lllyasviel
2024-01-25 08:33:57 -08:00
parent 43dabe808e
commit 222f77538e
2 changed files with 2 additions and 2 deletions

View File

@@ -1326,7 +1326,6 @@ 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()