From a2430fb880fcc6627a55a2bb97acbb4c58284e3f Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Thu, 14 Nov 2024 20:02:51 -0600 Subject: [PATCH] Add clarifying comment so I don't accidentally cause regression in the future with new memory management system --- adv_control/control.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/adv_control/control.py b/adv_control/control.py index b59d1a8..ee83072 100644 --- a/adv_control/control.py +++ b/adv_control/control.py @@ -129,7 +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() + to_return.control_model_wrapped = v.control_model_wrapped.clone() # needed to avoid breaking memory management system (parent tracking) return to_return