lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 18:51:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sd_models.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 07:24:15 -08:00
parent 232c90aba2
commit f75bf1ba50
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_models.py
View File

@@ -10,6 +10,7 @@ from omegaconf import OmegaConf, ListConfig
from os import mkdir from os import mkdir
from urllib import request from urllib import request
import ldm.modules.midas as midas import ldm.modules.midas as midas
import gc
from ldm.util import instantiate_from_config from ldm.util import instantiate_from_config
@@ -581,6 +582,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None):
model_data.sd_model = None model_data.sd_model = None
model_management.unload_all_models() model_management.unload_all_models()
model_management.soft_empty_cache() model_management.soft_empty_cache()
gc.collect()
timer.record("unload existing model") timer.record("unload existing model")