OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-03-09 23:44:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

minor bug fix of sd model memory management

Browse Source
This commit is contained in:
Boning 2024-03-21 15:28:38 -07:00
parent bef51aed03
commit f62217b65d
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
modules/sd_models.py
View File

@ -787,6 +787,13 @@ def reuse_model_from_already_loaded(sd_model, checkpoint_info, timer):
Additionaly deletes loaded models that are over the limit set in settings (sd_checkpoints_limit). Additionaly deletes loaded models that are over the limit set in settings (sd_checkpoints_limit).
""" """
if sd_model is not None and sd_model.sd_checkpoint_info.filename == checkpoint_info.filename:
return sd_model
if shared.opts.sd_checkpoints_keep_in_cpu:
send_model_to_cpu(sd_model)
timer.record("send model to cpu")
already_loaded = None already_loaded = None
for i in reversed(range(len(model_data.loaded_sd_models))): for i in reversed(range(len(model_data.loaded_sd_models))):
loaded_model = model_data.loaded_sd_models[i] loaded_model = model_data.loaded_sd_models[i]
@ -800,10 +807,6 @@ def reuse_model_from_already_loaded(sd_model, checkpoint_info, timer):
send_model_to_trash(loaded_model) send_model_to_trash(loaded_model)
timer.record("send model to trash") timer.record("send model to trash")
if shared.opts.sd_checkpoints_keep_in_cpu:
send_model_to_cpu(sd_model)
timer.record("send model to cpu")
if already_loaded is not None: if already_loaded is not None:
send_model_to_device(already_loaded) send_model_to_device(already_loaded)
timer.record("send model to device") timer.record("send model to device")