From f75bf1ba50bf75c2c4c0183ae69c08150afc10a4 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 07:24:15 -0800 Subject: [PATCH] Update sd_models.py --- modules/sd_models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules/sd_models.py b/modules/sd_models.py index b46c4a6f..6e961cbb 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -10,6 +10,7 @@ from omegaconf import OmegaConf, ListConfig from os import mkdir from urllib import request import ldm.modules.midas as midas +import gc 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_management.unload_all_models() model_management.soft_empty_cache() + gc.collect() timer.record("unload existing model")