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

Merge pull request #135 from DanaCase/test_low_ram

Browse Source 
Update arg parser test fixture
This commit is contained in:
Filip Strand 2025-03-05 07:23:51 +01:00 committed by GitHub
commit ceea386361
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/mflux/train.py
View File

@ -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()

2
tests/arg_parser/test_cli_argparser.py
View File

@ -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