InvokeAI/scripts/invokeai-cli.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

23 lines
528 B
Python
Raw Permalink Normal View History

#!/usr/bin/env python
# Copyright (c) 2022 Kyle Schouviller (https://github.com/kyle0654)
2023-05-12 14:13:49 +00:00
import logging
2023-08-18 15:13:28 +00:00
import os
2023-07-27 14:54:01 +00:00
2023-05-12 14:13:49 +00:00
logging.getLogger("xformers").addFilter(lambda record: "A matching Triton is not available" not in record.getMessage())
2023-07-27 14:54:01 +00:00
def main():
# Change working directory to the repo root
os.chdir(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
# TODO: Parse some top-level args here.
from invokeai.app.cli_app import invoke_cli
2023-07-27 14:54:01 +00:00
invoke_cli()
2023-07-27 14:54:01 +00:00
if __name__ == "__main__":
main()