diffusers/src/diffusers/commands/diffusers_cli.py at main · cryptexis/diffusers · GitHub

43 lines (33 loc) · 1.29 KB

#!/usr/bin/env python

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

# http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

from argparse import ArgumentParser

from .env import EnvironmentCommand

from .fp16_safetensors import FP16SafetensorsCommand

def main():

parser = ArgumentParser("Diffusers CLI tool", usage="diffusers-cli <command> [<args>]")

commands_parser = parser.add_subparsers(help="diffusers-cli command helpers")

# Register commands

EnvironmentCommand.register_subcommand(commands_parser)

FP16SafetensorsCommand.register_subcommand(commands_parser)

# Let's go

args = parser.parse_args()

if not hasattr(args, "func"):

parser.print_help()

exit(1)

# Run

service = args.func(args)

service.run()

if __name__ == "__main__":

main()

Expand file tree