From a942ca8d67647c01650fa368af517e5266cfe986 Mon Sep 17 00:00:00 2001 From: filipstrand Date: Mon, 23 Sep 2024 20:22:02 +0200 Subject: [PATCH] Rename controlnet path arg --- src/mflux/controlnet/flux_controlnet.py | 6 +++--- src/mflux/generate_controlnet.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/mflux/controlnet/flux_controlnet.py b/src/mflux/controlnet/flux_controlnet.py index ca73a8f..fb68368 100644 --- a/src/mflux/controlnet/flux_controlnet.py +++ b/src/mflux/controlnet/flux_controlnet.py @@ -108,7 +108,7 @@ class Flux1Controlnet: seed: int, prompt: str, output: str, - control_image_path: str, + controlnet_image_path: str, controlnet_save_canny: bool = False, config: ConfigControlnet = ConfigControlnet() ) -> "GeneratedImage": # fmt: off @@ -117,7 +117,7 @@ class Flux1Controlnet: time_steps = tqdm(range(config.num_inference_steps)) # Embedd the controlnet reference image - control_image = ImageUtil.load_image(control_image_path) + control_image = ImageUtil.load_image(controlnet_image_path) control_image = ControlnetUtil.scale_image(config.height, config.width, control_image) control_image = ControlnetUtil.preprocess_canny(control_image) if controlnet_save_canny: @@ -180,7 +180,7 @@ class Flux1Controlnet: lora_paths=self.lora_paths, lora_scales=self.lora_scales, config=config, - controlnet_image_path=control_image_path, + controlnet_image_path=controlnet_image_path, ) @staticmethod diff --git a/src/mflux/generate_controlnet.py b/src/mflux/generate_controlnet.py index 071f787..5a2d861 100644 --- a/src/mflux/generate_controlnet.py +++ b/src/mflux/generate_controlnet.py @@ -47,7 +47,7 @@ def main(): seed=int(time.time()) if args.seed is None else args.seed, prompt=args.prompt, output=args.output, - control_image_path=args.controlnet_image_path, + controlnet_image_path=args.controlnet_image_path, controlnet_save_canny=args.controlnet_save_canny, config=ConfigControlnet( num_inference_steps=args.steps,