From e4c7aa75f7e48dd312b28cd4f5cf1724d21a4257 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Wed, 16 Apr 2025 22:52:30 +0300 Subject: [PATCH] Fix balanced offload offload post not activating --- modules/sd_offload.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/sd_offload.py b/modules/sd_offload.py index 9352d4f38..c57165fb5 100644 --- a/modules/sd_offload.py +++ b/modules/sd_offload.py @@ -299,7 +299,7 @@ def apply_balanced_offload(sd_model=None, exclude=[]): if device_map and max_memory: module.balanced_offload_device_map = device_map module.balanced_offload_max_memory = max_memory - module.offload_post = shared.sd_model_type in [offload_post] and shared.opts.te_hijack and module_name.startswith("text_encoder") + module.offload_post = shared.sd_model_type in offload_post and shared.opts.te_hijack and module_name.startswith("text_encoder") devices.torch_gc(fast=True, force=True, reason='offload') apply_balanced_offload_to_module(sd_model)