Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

(installer) ask the user for their GPU type; improve other messaging

Browse Source
This commit is contained in:
Eugene Brodsky 2023-01-13 04:10:34 -05:00
parent 88526b9294
commit b186965e77
2 changed files with 49 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
installer/main.py
View File

@ -19,4 +19,4 @@ if __name__ == "__main__":
except KeyboardInterrupt as exc:
print("\n")
print("Ctrl-C pressed. Aborting.")
print("See you again soon!")
print("Come back soon!")

55
installer/messages.py
View File

@ -95,6 +95,7 @@ def dest_path(init_path=None) -> Path:
dest = dest.expanduser().resolve()
if dest.exists():
console.line()
print(f":exclamation: Directory {dest} already exists.")
console.line()
dest_confirmed = Confirm.ask(
@ -136,8 +137,8 @@ def graphical_accelerator():
print(f"Only CPU acceleration is available on {ARCH} architecture. Proceeding with that.")
return "cpu"
nvidia = ("an [dark_sea_green4 b]NVIDIA[/] GPU (using CUDA™). [steel_blue3]https://link.to.supported.GPUs[/])", "cuda",)
amd = ("an [red3 b]AMD[/] GPU (using ROCm™). [steel_blue3]https://link.to.supported.GPUs[/])", "rocm",)
nvidia = ("an [gold1 b]NVIDIA[/] GPU (using CUDA™)", "cuda",)
amd = ("an [gold1 b]AMD[/] GPU (using ROCm™)", "rocm",)
cpu = ("no compatible GPU, or specifically prefer to use the CPU", "cpu",)
idk = ("I'm not sure what to choose", "idk",)
@ -158,17 +159,57 @@ def graphical_accelerator():
options = {str(i): opt for i, opt in enumerate(options, 1)}
console.rule("GPU (Graphics Card) selection")
console.rule(":space_invader: GPU (Graphics Card) selection :space_invader:")
console.print(Panel(
Group(
f"Detected the [gold1]{OS}-{ARCH}[/] platform. Please select the installation option for GPU support",
Group("\n".join([
f"Detected the [gold1]{OS}-{ARCH}[/] platform",
"",
Panel("\n".join([f"[green b i]{i}[/] [dark_red]🢒[/] {opt[0]}" for (i, opt) in options.items()]), box=box.MINIMAL),
"See [steel_blue3]https://invoke-ai.github.io/InvokeAI/#system[/] to ensure your system meets the minimum requirements.",
"",
"[red3]🠶[/] [b]Your GPU drivers must be correctly installed before using InvokeAI![/] [red3]🠴[/]"]),
"",
"Please select the type of GPU installed in your computer.",
Panel("\n".join([f"[spring_green3 b i]{i}[/] [dark_red]🢒[/]{opt[0]}" for (i, opt) in options.items()]), box=box.MINIMAL),
),
box=box.MINIMAL,
padding=(1, 1)
padding=(1, 1),
)
)
choice = prompt("Please make your selection: ", validator=Validator.from_callable(lambda n: n in options.keys(), error_message="Please select one the above options"))
if options[choice][1] == "idk":
console.print("No problem. We will try to install a version that [i]should[/i] be compatible. :crossed_fingers:")
return options[choice][1]
def simple_banner(message: str) -> None:
"""
A simple banner with a message, defined here for styling consistency
:param message: The message to display
:type message: str
"""
console.rule(message)
def introduction() -> None:
"""
Display a banner when starting configuration of the InvokeAI application
"""
console.rule()
console.print(Panel(title=":art: Configuring InvokeAI :art:", renderable=Group(
"",
"[b]This script will:",
"",
"1. Configure the InvokeAI application directory",
"2. Help download the Stable Diffusion weight files",
" and other large models that are needed for text to image generation",
"3. Create initial configuration files.",
"",
"[i]At any point you may interrupt this program and resume later.",
)))
console.line(2)