Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 10:35:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Installer: add option to install requirements for just one extension

Browse source
This commit is contained in:
oobabooga 2024-03-06 07:36:23 -08:00
parent 2174958362
commit fcc92caa30
1 changed files with 46 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

65
one_click.py
View file

@ -200,6 +200,16 @@ def run_cmd(cmd, assert_success=False, environment=False, capture_output=False,
return result return result
def generate_alphabetic_sequence(index):
result = ''
while index >= 0:
index, remainder = divmod(index, 26)
result = chr(ord('A') + remainder) + result
index -= 1
return result
def get_user_choice(question, options_dict): def get_user_choice(question, options_dict):
print() print()
print(question) print(question)
@ -308,9 +318,13 @@ def install_webui():
update_requirements(initial_installation=True) update_requirements(initial_installation=True)
def get_extensions_names():
return [foldername for foldername in os.listdir('extensions') if os.path.isfile(os.path.join('extensions', foldername, 'requirements.txt'))]
def install_extensions_requirements(): def install_extensions_requirements():
print_big_message("Installing extensions requirements.\nSome of these may fail on Windows.\nDon\'t worry if you see error messages, as they will not affect the main program.") print_big_message("Installing extensions requirements.\nSome of these may fail on Windows.\nDon\'t worry if you see error messages, as they will not affect the main program.")
extensions = [foldername for foldername in os.listdir('extensions') if os.path.isfile(os.path.join('extensions', foldername, 'requirements.txt'))] extensions = get_extensions_names()
for i, extension in enumerate(extensions): for i, extension in enumerate(extensions):
print(f"\n\n--- [{i+1}/{len(extensions)}]: {extension}\n\n") print(f"\n\n--- [{i+1}/{len(extensions)}]: {extension}\n\n")
extension_req_path = os.path.join("extensions", extension, "requirements.txt") extension_req_path = os.path.join("extensions", extension, "requirements.txt")
@ -414,25 +428,38 @@ if __name__ == "__main__":
args, _ = parser.parse_known_args() args, _ = parser.parse_known_args()
if args.update_wizard: if args.update_wizard:
choice = get_user_choice( while True:
"What would you like to do?", choice = get_user_choice(
{ "What would you like to do?",
'A': 'Update the web UI', {
'B': 'Install/update extensions requirements', 'A': 'Update the web UI',
'C': 'Revert local changes to repository files with \"git reset --hard\"', 'B': 'Install/update extensions requirements',
'N': 'Nothing (exit)' 'C': 'Revert local changes to repository files with \"git reset --hard\"',
}, 'N': 'Nothing (exit)'
) },
)
if choice == 'A': if choice == 'A':
update_requirements() update_requirements()
elif choice == 'B': elif choice == 'B':
install_extensions_requirements() choices = {'A': 'All extensions'}
update_requirements(pull=False) for i, name in enumerate(get_extensions_names()):
elif choice == 'C': key = generate_alphabetic_sequence(i + 1)
run_cmd("git reset --hard", assert_success=True, environment=True) choices[key] = name
elif choice == 'N':
sys.exit() choice = get_user_choice("What extension?", choices)
if choice == 'A':
install_extensions_requirements()
else:
extension_req_path = os.path.join("extensions", choices[choice], "requirements.txt")
run_cmd(f"python -m pip install -r {extension_req_path} --upgrade", assert_success=False, environment=True)
update_requirements(pull=False)
elif choice == 'C':
run_cmd("git reset --hard", assert_success=True, environment=True)
elif choice == 'N':
sys.exit()
else: else:
if not is_installed(): if not is_installed():
install_webui() install_webui()