lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-09 01:19:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update processing.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 08:28:20 -08:00
parent 9ab5589a62
commit 43dabe808e
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/processing.py
View File

@@ -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()