def register_subcommand()

in src/autotrain/cli/run_tools.py [0:0]

• 57 lines of code
• 1 McCabe index (conditional complexity)

    def register_subcommand(parser: ArgumentParser):
        run_app_parser = parser.add_parser("tools", help="Run AutoTrain tools")
        subparsers = run_app_parser.add_subparsers(title="tools", dest="tool_name")

        merge_llm_parser = subparsers.add_parser(
            "merge-llm-adapter",
            help="Merge LLM Adapter tool",
        )
        merge_llm_parser.add_argument(
            "--base-model-path",
            type=str,
            help="Base model path",
        )
        merge_llm_parser.add_argument(
            "--adapter-path",
            type=str,
            help="Adapter path",
        )
        merge_llm_parser.add_argument(
            "--token",
            type=str,
            help="Token",
            default=None,
            required=False,
        )
        merge_llm_parser.add_argument(
            "--pad-to-multiple-of",
            type=int,
            help="Pad to multiple of",
            default=None,
            required=False,
        )
        merge_llm_parser.add_argument(
            "--output-folder",
            type=str,
            help="Output folder",
            required=False,
            default=None,
        )
        merge_llm_parser.add_argument(
            "--push-to-hub",
            action="store_true",
            help="Push to Hugging Face Hub",
            required=False,
        )
        merge_llm_parser.set_defaults(func=run_tools_command_factory, merge_llm_adapter=True)

        convert_to_kohya_parser = subparsers.add_parser("convert_to_kohya", help="Convert to Kohya tool")
        convert_to_kohya_parser.add_argument(
            "--input-path",
            type=str,
            help="Input path",
        )
        convert_to_kohya_parser.add_argument(
            "--output-path",
            type=str,
            help="Output path",
        )
        convert_to_kohya_parser.set_defaults(func=run_tools_command_factory, convert_to_kohya=True)