feat(nodes): add separate scripts to launch cli and web

This commit is contained in:
psychedelicious 2023-06-04 22:00:57 +10:00
parent ff204db6b2
commit 383e3d77cb
3 changed files with 23 additions and 12 deletions

View File

@ -1,5 +0,0 @@
#!/usr/bin/env python
from invokeai.frontend.CLI import invokeai_command_line_interface as main
main()

View File

@ -12,13 +12,9 @@ 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()
# TODO: Parse some top-level args here.
from invokeai.app.cli_app import invoke_cli
invoke_cli()
if __name__ == '__main__':

20
scripts/invokeai-web.py Executable file
View File

@ -0,0 +1,20 @@
#!/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__), "..")))
from invokeai.app.api_app import invoke_api
invoke_api()
if __name__ == '__main__':
main()