lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-02 14:27:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update controlnet.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-29 16:47:56 -08:00
parent be2d7edaee
commit f5b218e264
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
View File

@@ -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):