hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 19:22:32 +00:00
Code Issues Actions Packages Projects Releases 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