From 1de1759203871bafb86ffe3b3a8d02c29db372f6 Mon Sep 17 00:00:00 2001 From: Boning Date: Thu, 21 Mar 2024 15:28:38 -0700 Subject: [PATCH] minor bug fix of sd model memory management --- modules/sd_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 747fc39ee69..b922638dd29 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -801,7 +801,7 @@ def reuse_model_from_already_loaded(sd_model, checkpoint_info, timer): timer.record("send model to trash") if shared.opts.sd_checkpoints_keep_in_cpu: - send_model_to_cpu(sd_model) + send_model_to_cpu(loaded_model) timer.record("send model to cpu") if already_loaded is not None: