mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-06 10:31:17 -08:00
Code Issues Projects Releases Packages Wiki Activity

repair unload sd checkpoint button

Browse source
This commit is contained in:
AUTOMATIC1111 2023-10-15 09:41:02 +03:00
parent 0d65d0eabd
commit 282903bb67
3 changed files with 23 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

13
modules/sd_models.py
View file

@ -1,7 +1,6 @@
import collections
import os.path
import sys
import gc
import threading
import torch
@ -798,17 +797,7 @@ def reload_model_weights(sd_model=None, info=None):
def unload_model_weights(sd_model=None, info=None):
timer = Timer()
if model_data.sd_model:
model_data.sd_model.to(devices.cpu)
sd_hijack.model_hijack.undo_hijack(model_data.sd_model)
model_data.sd_model = None
sd_model = None
gc.collect()
devices.torch_gc()
print(f"Unloaded weights {timer.summary()}.")
send_model_to_cpu(sd_model or shared.sd_model)
return sd_model