AUTOMATIC1111/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-03-13 00:50:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #15600 from AUTOMATIC1111/fix-corrupt-model-loop

Browse Source 
Fix corrupt model initial load loop
This commit is contained in:
AUTOMATIC1111
2024-06-08 10:12:45 +03:00
committed by GitHub
parent 5429e4cff5 4bc39d234d
commit 15245d9d5e
2 changed files with 17 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modules/sd_models.py
View File

@@ -659,10 +659,11 @@ def get_empty_cond(sd_model):
def send_model_to_cpu(m):
if m.lowvram:
lowvram.send_everything_to_cpu()
else:
m.to(devices.cpu)
if m is not None:
if m.lowvram:
lowvram.send_everything_to_cpu()
else:
m.to(devices.cpu)
devices.torch_gc()