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

frontend design done; functionality not hooked up yet

Browse Source
This commit is contained in:
Lincoln Stein
2023-02-14 00:02:19 -05:00
parent 9cacba916b
commit 7545e38655
6 changed files with 84 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
ldm/invoke/config/invokeai_configure.py
View File

@ -31,10 +31,9 @@ from transformers import (
)
import invokeai.configs as configs
from ldm.invoke.config.initial_model_select import (
from ldm.invoke.config.model_install import (
download_from_hf,
select_and_download_models,
yes_or_no,
)
from ldm.invoke.globals import Globals, global_config_dir
from ldm.invoke.readline import generic_completer
@ -103,6 +102,18 @@ Have fun!
print(message)
# ---------------------------------------------
def yes_or_no(prompt: str, default_yes=True):
completer.set_options(["yes", "no"])
completer.complete_extensions(None) # turn off path-completion mode
default = "y" if default_yes else "n"
response = input(f"{prompt} [{default}] ") or default
if default_yes:
return response[0] not in ("n", "N")
else:
return response[0] in ("y", "Y")
# ---------------------------------------------
def HfLogin(access_token) -> str:
"""