From a9e9ec7b929aa769761565a1021c0312580f95d1 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Fri, 30 Aug 2024 03:28:01 -0500 Subject: [PATCH] Fix inconsistent variable name causing reference before assignment error --- 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 4e4c5d7..5922e45 100644 --- a/adv_control/control.py +++ b/adv_control/control.py @@ -107,7 +107,7 @@ class ControlNetAdvanced(ControlNet, AdvancedControlBase): self.is_flux = "Flux" in str(type(self.control_model).__name__) return super().pre_run_advanced(*args, **kwargs) - def apply_advanced_strengths_and_masks(self, x: Tensor, batched_number: int, is_flux=False): + def apply_advanced_strengths_and_masks(self, x: Tensor, batched_number: int, flux_shape=None): if self.is_flux: flux_shape = self.x_noisy_shape return super().apply_advanced_strengths_and_masks(x, batched_number, flux_shape)