fix refiner (#2042)

scripts/controlnet.py

@@ -8,6 +8,7 @@
 import modules.scripts as scripts
 from modules import shared, devices, script_callbacks, processing, masking, images
 import gradio as gr
+import time
 
 
 from einops import rearrange
@@ -636,12 +637,7 @@ def bound_check_params(unit: external_code.ControlNetUnit) -> None:
                 setattr(unit, param, default_value)
                 logger.warning(f'[{unit.module}.{param}] Invalid value({value}), using default value {default_value}.')
 
-    def process(self, p, *args):
-        """
-        This function is called before processing begins for AlwaysVisible scripts.
-        You can modify the processing object (p) here, inject hooks, etc.
-        args contains all values returned by components from ui()
-        """
+    def controlnet_main_entry(self, p):
         sd_ldm = p.sd_model
         unet = sd_ldm.model.diffusion_model
         self.noise_modifier = None
@@ -980,6 +976,10 @@ def before_process_batch(self, p, *args, **kwargs):
                                    noise_modifier=self.noise_modifier,
                                    sd_model=p.sd_model)
         self.noise_modifier = None
+
+        t = time.time()
+        self.controlnet_main_entry(p)
+        logger.info(f'ControlNet Hooked - Time = {time.time() - t}')
         return
 
     def postprocess_batch(self, p, *args, **kwargs):

scripts/controlnet_version.py

@@ -1,4 +1,4 @@
-version_flag = 'v1.1.400'
+version_flag = 'v1.1.401'
 
 from scripts.logging import logger