From b9c53b79c9e177efebe0f4f3197ac22e76ef9c45 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Thu, 14 Nov 2024 11:49:56 -0600 Subject: [PATCH] Fix from_vanilla call for ControlNetAdvanced so it does not break ComfyUI memory management system; parent model patcher has to be tracked properly now --- adv_control/control.py | 1 + 1 file changed, 1 insertion(+) diff --git a/adv_control/control.py b/adv_control/control.py index c589ed1..b59d1a8 100644 --- a/adv_control/control.py +++ b/adv_control/control.py @@ -129,6 +129,7 @@ class ControlNetAdvanced(ControlNet, AdvancedControlBase): global_average_pooling=v.global_average_pooling, compression_ratio=v.compression_ratio, latent_format=v.latent_format, load_device=v.load_device, manual_cast_dtype=v.manual_cast_dtype) v.copy_to(to_return) + to_return.control_model_wrapped = v.control_model_wrapped.clone() return to_return