mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-02-03 06:13:06 -08:00
Code Issues Projects Releases Packages 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
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:
p.scripts.postprocess(p, res)
if lowvram.is_enabled(shared.sd_model):
# for interrupted case
lowvram.send_everything_to_cpu()
return res