cxllm-vendors/stable-diffusion-webui-forge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lllyasviel
2024-01-30 14:40:26 -08:00
parent b4dafc07b4
commit 2336ec7da8
2 changed files with 21 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
+5 -5
View File
@@ -529,8 +529,8 @@ class ControlNetForForgeOfficial(scripts.Script):
params.model.positive_advanced_weighting = soft_weighting.copy()
params.model.negative_advanced_weighting = soft_weighting.copy()
params.preprocessor.process_before_every_sampling(process=p, cond=cond, **kwargs)
params.model.process_before_every_sampling(process=p, cond=cond, **kwargs)
params.preprocessor.process_before_every_sampling(p, cond, *args, **kwargs)
params.model.process_before_every_sampling(p, cond, *args, **kwargs)
logger.info(f"ControlNet Method {params.preprocessor.name} patched.")
return
@@ -542,8 +542,8 @@ class ControlNetForForgeOfficial(scripts.Script):
params: ControlNetCachedParameters,
*args, **kwargs):
params.preprocessor.process_after_every_sampling(process=p, params=params, **kwargs)
params.model.process_after_every_sampling(process=p, params=params, **kwargs)
params.preprocessor.process_after_every_sampling(p, params, *args, **kwargs)
params.model.process_after_every_sampling(p, params, *args, **kwargs)
return
def process(self, p, *args, **kwargs):
@@ -560,7 +560,7 @@ class ControlNetForForgeOfficial(scripts.Script):
self.process_unit_before_every_sampling(p, unit, self.current_params[i], *args, **kwargs)
return
def postprocess_batch(self, p, *args, **kwargs):
def postprocess_batch_list(self, p, *args, **kwargs):
for i, unit in enumerate(self.get_enabled_units(p)):
self.process_unit_after_every_sampling(p, unit, self.current_params[i], *args, **kwargs)
self.current_params = {}