diff --git a/src/mflux/generate_controlnet.py b/src/mflux/generate_controlnet.py index 257b806..0e077f3 100644 --- a/src/mflux/generate_controlnet.py +++ b/src/mflux/generate_controlnet.py @@ -8,7 +8,8 @@ def main(): # fmt: off parser = argparse.ArgumentParser(description="Generate an image based on a prompt.") parser.add_argument("--prompt", type=str, required=True, help="The textual description of the image to generate.") - parser.add_argument("--control-image-path", type=str, required=True, help="Local path of the image to use as input for controlnet.") + parser.add_argument("--controlnet-image-path", type=str, required=True, help="Local path of the image to use as input for controlnet.") + parser.add_argument("--controlnet-strength", type=float, default=0.4, help="Controls how strongly the control image influences the output image. A value of 0.0 means no influence. (Default is 0.4)") parser.add_argument("--output", type=str, default="image.png", help="The filename for the output image. Default is \"image.png\".") parser.add_argument("--model", "-m", type=str, required=True, choices=["dev", "schnell"], help="The model to use (\"schnell\" or \"dev\").") parser.add_argument("--seed", type=int, default=None, help="Entropy Seed (Default is time-based random-seed)") @@ -16,7 +17,6 @@ def main(): parser.add_argument("--width", type=int, default=1024, help="Image width (Default is 1024)") parser.add_argument("--steps", type=int, default=None, help="Inference Steps") parser.add_argument("--guidance", type=float, default=3.5, help="Guidance Scale (Default is 3.5)") - parser.add_argument("--controlnet-strength", type=float, default=0.7, help="Controls how strongly the control image influences the output image. A value of 0.0 means no influence. (Default is 0.7)") parser.add_argument("--quantize", "-q", type=int, choices=[4, 8], default=None, help="Quantize the model (4 or 8, Default is None)") parser.add_argument("--path", type=str, default=None, help="Local path for loading a model from disk") parser.add_argument("--lora-paths", type=str, nargs="*", default=None, help="Local safetensors for applying LORA from disk") @@ -45,7 +45,7 @@ def main(): image = flux.generate_image( seed=int(time.time()) if args.seed is None else args.seed, prompt=args.prompt, - control_image=ImageUtil.load_image(args.control_image_path), + control_image=ImageUtil.load_image(args.controlnet_image_path), config=ConfigControlnet( num_inference_steps=args.steps, height=args.height,