Rename save arg

src/mflux/controlnet/flux_controlnet.py

@@ -109,7 +109,7 @@ class Flux1Controlnet:
             prompt: str,
             output: str,
             control_image_path: str,
-            save_control_image_canny: bool = False,
+            controlnet_save_canny: bool = False,
             config: ConfigControlnet = ConfigControlnet()
     ) -> "GeneratedImage":  # fmt: off
         # Create a new runtime config based on the model type and input parameters
@@ -120,7 +120,7 @@ class Flux1Controlnet:
         control_image = ImageUtil.load_image(control_image_path)
         control_image = ControlnetUtil.scale_image(config.height, config.width, control_image)
         control_image = ControlnetUtil.preprocess_canny(control_image)
-        if save_control_image_canny:
+        if controlnet_save_canny:
             ControlnetUtil.save_canny_image(control_image, output)
         controlnet_cond = ImageUtil.to_array(control_image)
         controlnet_cond = self.vae.encode(controlnet_cond)

src/mflux/generate_controlnet.py

@@ -48,7 +48,7 @@ def main():
         prompt=args.prompt,
         output=args.output,
         control_image_path=args.controlnet_image_path,
-        save_control_image_canny=args.controlnet_save_canny,
+        controlnet_save_canny=args.controlnet_save_canny,
         config=ConfigControlnet(
             num_inference_steps=args.steps,
             height=args.height,