From bf1e366e438fe267850170b402201fe6502d3a65 Mon Sep 17 00:00:00 2001 From: Dana Case Date: Mon, 3 Mar 2025 18:54:49 -0800 Subject: [PATCH 1/2] formatting --- tests/arg_parser/test_cli_argparser.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/arg_parser/test_cli_argparser.py b/tests/arg_parser/test_cli_argparser.py index 8ea9d9a..56bee4d 100644 --- a/tests/arg_parser/test_cli_argparser.py +++ b/tests/arg_parser/test_cli_argparser.py @@ -10,6 +10,7 @@ from mflux.ui.cli.parsers import CommandLineParser def _create_mflux_generate_parser(with_controlnet=False) -> CommandLineParser: parser = CommandLineParser(description="Generate an image based on a prompt.") + parser.add_general_arguments() parser.add_model_arguments(require_model_arg=False) parser.add_image_generator_arguments(supports_metadata_config=True) parser.add_lora_arguments() @@ -33,6 +34,7 @@ def mflux_generate_controlnet_parser() -> CommandLineParser: @pytest.fixture def mflux_save_parser() -> CommandLineParser: parser = CommandLineParser(description="Save a quantized version of Flux.1 to disk.") # fmt: off + parser.add_general_arguments() parser.add_model_arguments(path_type="save", require_model_arg=True) parser.add_lora_arguments() return parser From 50734795b93919d25cfe7159836c6309da8b25af Mon Sep 17 00:00:00 2001 From: Dana Case Date: Mon, 3 Mar 2025 19:21:01 -0800 Subject: [PATCH 2/2] add general arguments to train as well --- src/mflux/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/mflux/train.py b/src/mflux/train.py index 1d035c1..a8f07ae 100644 --- a/src/mflux/train.py +++ b/src/mflux/train.py @@ -6,6 +6,7 @@ from mflux.ui.cli.parsers import CommandLineParser def main(): parser = CommandLineParser(description="Finetune a LoRA adapter") + parser.add_general_arguments() parser.add_model_arguments(require_model_arg=False) parser.add_training_arguments() args = parser.parse_args()