From ec8b42e5215ce95ef5b736265ce9afc42ce15524 Mon Sep 17 00:00:00 2001 From: filipstrand Date: Mon, 23 Sep 2024 20:20:59 +0200 Subject: [PATCH] Rename save arg --- src/mflux/controlnet/flux_controlnet.py | 4 ++-- src/mflux/generate_controlnet.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/mflux/controlnet/flux_controlnet.py b/src/mflux/controlnet/flux_controlnet.py index 1483dd0..ca73a8f 100644 --- a/src/mflux/controlnet/flux_controlnet.py +++ b/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) diff --git a/src/mflux/generate_controlnet.py b/src/mflux/generate_controlnet.py index b1ecfaf..071f787 100644 --- a/src/mflux/generate_controlnet.py +++ b/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,