()
| 20 | |
| 21 | |
| 22 | def main(): |
| 23 | parser = ArgumentParser("Diffusers CLI tool", usage="diffusers-cli <command> [<args>]") |
| 24 | commands_parser = parser.add_subparsers(help="diffusers-cli command helpers") |
| 25 | |
| 26 | # Register commands |
| 27 | EnvironmentCommand.register_subcommand(commands_parser) |
| 28 | FP16SafetensorsCommand.register_subcommand(commands_parser) |
| 29 | |
| 30 | # Let's go |
| 31 | args = parser.parse_args() |
| 32 | |
| 33 | if not hasattr(args, "func"): |
| 34 | parser.print_help() |
| 35 | exit(1) |
| 36 | |
| 37 | # Run |
| 38 | service = args.func(args) |
| 39 | service.run() |
| 40 | |
| 41 | |
| 42 | if __name__ == "__main__": |
no test coverage detected