From f5b218e26467c1c1930fbc24f7c60de3cdd17011 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Mon, 29 Jan 2024 16:47:56 -0800 Subject: [PATCH] Update controlnet.py --- extensions-builtin/sd_forge_controlnet/scripts/controlnet.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py index 1662610c..ea33338b 100644 --- a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py +++ b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py @@ -473,6 +473,7 @@ class ControlNetForForgeOfficial(scripts.Script): params.preprocessor.process_before_every_sampling(process=p, cond=cond, **kwargs) params.model.process_before_every_sampling(process=p, cond=cond, **kwargs) + logger.info(f"ControlNet Method {params.preprocessor.name} patched.") return def process(self, p, *args, **kwargs): @@ -485,8 +486,8 @@ class ControlNetForForgeOfficial(scripts.Script): return def process_before_every_sampling(self, p, *args, **kwargs): - for unit, params in zip(self.get_enabled_units(p), self.current_params): - self.process_unit_before_every_sampling(p, unit, params, *args, **kwargs) + for i, unit in enumerate(self.get_enabled_units(p)): + self.process_unit_before_every_sampling(p, unit, self.current_params[i], *args, **kwargs) return def postprocess(self, p, processed, *args):