From bf1e366e438fe267850170b402201fe6502d3a65 Mon Sep 17 00:00:00 2001 From: Dana Case Date: Mon, 3 Mar 2025 18:54:49 -0800 Subject: [PATCH] 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