From ba03d74920337680aab3769a6974a4e08fdda1e1 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sat, 27 Jan 2024 22:13:29 -0800 Subject: [PATCH] Update forge_sampler.py --- modules_forge/forge_sampler.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/modules_forge/forge_sampler.py b/modules_forge/forge_sampler.py index ea6ea60f..5ffb9795 100644 --- a/modules_forge/forge_sampler.py +++ b/modules_forge/forge_sampler.py @@ -77,8 +77,12 @@ def sampling_prepare(unet, x): B, C, H, W = x.shape unet_inference_memory = unet.memory_required([B * 2, C, H, W]) - additional_inference_memory = unet.controlnet_linked_list.inference_memory_requirements(unet.model_dtype()) - additional_model_patchers = unet.get_models() + additional_inference_memory = 0 + additional_model_patchers = [] + + if unet.controlnet_linked_list is not None: + additional_inference_memory += unet.controlnet_linked_list.inference_memory_requirements(unet.model_dtype()) + additional_model_patchers += unet.controlnet_linked_list.get_models() model_management.load_models_gpu( models=[unet] + additional_model_patchers,