monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rename controlnet path arg

Browse Source
This commit is contained in:
filipstrand 2024-09-23 20:22:02 +02:00
parent ec8b42e521
commit a942ca8d67
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/mflux/controlnet/flux_controlnet.py
View File

@ -108,7 +108,7 @@ class Flux1Controlnet:
seed: int, seed: int,
prompt: str, prompt: str,
output: str, output: str,
control_image_path: str, controlnet_image_path: str,
controlnet_save_canny: bool = False, controlnet_save_canny: bool = False,
config: ConfigControlnet = ConfigControlnet() config: ConfigControlnet = ConfigControlnet()
) -> "GeneratedImage": # fmt: off ) -> "GeneratedImage": # fmt: off
@ -117,7 +117,7 @@ class Flux1Controlnet:
time_steps = tqdm(range(config.num_inference_steps)) time_steps = tqdm(range(config.num_inference_steps))
# Embedd the controlnet reference image # 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.scale_image(config.height, config.width, control_image)
control_image = ControlnetUtil.preprocess_canny(control_image) control_image = ControlnetUtil.preprocess_canny(control_image)
if controlnet_save_canny: if controlnet_save_canny:
@ -180,7 +180,7 @@ class Flux1Controlnet:
lora_paths=self.lora_paths, lora_paths=self.lora_paths,
lora_scales=self.lora_scales, lora_scales=self.lora_scales,
config=config, config=config,
controlnet_image_path=control_image_path, controlnet_image_path=controlnet_image_path,
) )
@staticmethod @staticmethod

2
src/mflux/generate_controlnet.py
View File

@ -47,7 +47,7 @@ def main():
seed=int(time.time()) if args.seed is None else args.seed, seed=int(time.time()) if args.seed is None else args.seed,
prompt=args.prompt, prompt=args.prompt,
output=args.output, output=args.output,
control_image_path=args.controlnet_image_path, controlnet_image_path=args.controlnet_image_path,
controlnet_save_canny=args.controlnet_save_canny, controlnet_save_canny=args.controlnet_save_canny,
config=ConfigControlnet( config=ConfigControlnet(
num_inference_steps=args.steps, num_inference_steps=args.steps,