Fix infotext (#28)

extensions-builtin/sd_forge_controlnet/lib_controlnet/infotext.py

@@ -32,11 +32,10 @@ def serialize_unit(unit: external_code.ControlNetUnit) -> str:
     excluded_fields = (
         "image",
         "enabled",
-        # Note: "advanced_weighting" is excluded as it is an API-only field.
-        "advanced_weighting",
-        # Note: "inpaint_crop_image" is img2img inpaint only flag, which does not
-        # provide much information when restoring the unit.
-        "inpaint_crop_input_image",
+        "input_mode",
+        "use_preview_as_input",
+        "generated_image",
+        "mask_image",
     )
 
     log_value = {

extensions-builtin/sd_forge_controlnet/scripts/controlnet.py

@@ -396,7 +396,9 @@ class ControlNetForForgeOfficial(scripts.Script):
     @torch.no_grad()
     def process(self, p, *args, **kwargs):
         self.current_params = {}
-        for i, unit in enumerate(self.get_enabled_units(args)):
+        enabled_units = self.get_enabled_units(args)
+        Infotext.write_infotext(enabled_units, p)
+        for i, unit in enumerate(enabled_units):
             self.bound_check_params(unit)
             params = ControlNetCachedParameters()
             self.process_unit_after_click_generate(p, unit, params, *args, **kwargs)