mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
58 lines
1.3 KiB
Python
58 lines
1.3 KiB
Python
"""
|
|
InvokeAI Installer
|
|
"""
|
|
|
|
import argparse
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from installer import Installer
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument(
|
|
"-r",
|
|
"--root",
|
|
dest="root",
|
|
type=str,
|
|
help="Destination path for installation",
|
|
default=os.environ.get("INVOKEAI_ROOT") or "~/invokeai",
|
|
)
|
|
parser.add_argument(
|
|
"-y",
|
|
"--yes",
|
|
"--yes-to-all",
|
|
dest="yes_to_all",
|
|
action="store_true",
|
|
help="Assume default answers to all questions",
|
|
default=False,
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--find-links",
|
|
dest="find_links",
|
|
help="Specifies a directory of local wheel files to be searched prior to searching the online repositories.",
|
|
type=Path,
|
|
default=None,
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--wheel",
|
|
dest="wheel",
|
|
help="Specifies a wheel for the InvokeAI package. Used for troubleshooting or testing prereleases.",
|
|
type=Path,
|
|
default=None,
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
inst = Installer()
|
|
|
|
try:
|
|
inst.install(**args.__dict__)
|
|
except KeyboardInterrupt:
|
|
print("\n")
|
|
print("Ctrl-C pressed. Aborting.")
|
|
print("Come back soon!")
|