OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-22 14:15:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

additional memory improvements when switching between models of different types

Browse Source
This commit is contained in:
AUTOMATIC1111 2023-07-31 10:43:31 +03:00
parent 29d7e31d89
commit 4d9b096663
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/sd_models.py
View File

@ -582,7 +582,10 @@ def reload_model_weights(sd_model=None, info=None):
timer.record("find config") timer.record("find config")
if sd_model is None or checkpoint_config != sd_model.used_config: if sd_model is None or checkpoint_config != sd_model.used_config:
del sd_model if sd_model is not None:
sd_model.to(device="meta")
devices.torch_gc()
load_model(checkpoint_info, already_loaded_state_dict=state_dict) load_model(checkpoint_info, already_loaded_state_dict=state_dict)
return model_data.sd_model return model_data.sd_model