InvokeAI/scripts/invoke-new.py

#!/usr/bin/env python

# Copyright (c) 2022 Kyle Schouviller (https://github.com/kyle0654)

import logging
logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())

import os
import sys

def main():
    # Change working directory to the repo root
    os.chdir(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))

    if '--web' in sys.argv:
        from invokeai.app.api_app import invoke_api
        invoke_api()
    else:
        # TODO: Parse some top-level args here.
        from invokeai.app.cli_app import invoke_cli
        invoke_cli()


if __name__ == '__main__':
    main()