diff --git a/modules/sd_models.py b/modules/sd_models.py index c8d15e2a..942cd7c4 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -587,8 +587,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None): model_data.sd_model = None model_management.unload_all_models() - - devices.torch_gc() + model_management.soft_empty_cache() timer.record("unload existing model")