more api

2024-01-29 23:28:13 -08:00
parent 8ee804a159
commit a1d2a31233
4 changed files with 16 additions and 8 deletions
@@ -460,6 +460,9 @@ class ControlNetForForgeOfficial(scripts.Script):
        params.model = cached_controlnet_loader(model_filename)
        params.preprocessor = preprocessor

+        params.preprocessor.process_after_running_preprocessors(process=p, params=params, **kwargs)
+        params.model.process_after_running_preprocessors(process=p, params=params, **kwargs)
+
        logger.info(f"Current ControlNet {type(params.model).__name__}: {model_filename}")
        return