Skip to content

Commit 47a3c64

Browse files
terribilissimoAndrea
andauthored
feat: expose img2img CLI flags for Flux.2 Klein (#358)
Co-authored-by: Andrea <p@styr.local>
1 parent f236371 commit 47a3c64

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/mflux/models/flux2/cli/flux2_generate.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ def main():
1616
parser.add_model_arguments(require_model_arg=False)
1717
parser.add_lora_arguments()
1818
parser.add_image_generator_arguments(supports_metadata_config=True, supports_dimension_scale_factor=True)
19+
parser.add_image_to_image_arguments(required=False)
1920
parser.add_output_arguments()
2021
args = parser.parse_args()
2122

@@ -49,7 +50,7 @@ def main():
4950
width, height = DimensionResolver.resolve(
5051
width=args.width,
5152
height=args.height,
52-
reference_image_path=None,
53+
reference_image_path=args.image_path,
5354
)
5455

5556
for seed in args.seed:
@@ -59,7 +60,9 @@ def main():
5960
width=width,
6061
height=height,
6162
guidance=args.guidance,
63+
image_path=args.image_path,
6264
num_inference_steps=args.steps,
65+
image_strength=args.image_strength,
6366
scheduler="flow_match_euler_discrete",
6467
)
6568
ImageUtil.save_image(

0 commit comments

Comments
 (0)