OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-01 19:22:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

call lowvram.send_everything_to_cpu() for interrupted case

Browse Source
This commit is contained in:
Won-Kyu Park 2024-09-19 10:24:59 +09:00
parent 1f779226f0
commit 380e9a84c3
No known key found for this signature in database
GPG Key ID: 53AA79C8C9535D15
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/processing.py
View File

@ -1147,6 +1147,11 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
if p.scripts is not None: if p.scripts is not None:
p.scripts.postprocess(p, res) p.scripts.postprocess(p, res)
if lowvram.is_enabled(shared.sd_model):
# for interrupted case
lowvram.send_everything_to_cpu()
return res return res