AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix extensions install

Browse Source
This commit is contained in:
oobabooga 2023-09-28 13:54:36 -07:00
parent e145d9a0da
commit 85f45cafa1
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
one_click.py
View File

@ -190,19 +190,18 @@ def update_requirements(initial_installation=False):
# Extensions requirements are installed only during the initial install by default.
# That can be changed with the INSTALL_EXTENSIONS environment variable.
install_extensions = os.environ.get("INSTALL_EXTENSIONS", "false").lower() in ("yes", "y", "true", "1", "t", "on")
if initial_installation or install_extensions:
if not install_extensions:
print_big_message("Will not install extensions due to INSTALL_EXTENSIONS environment variable.")
else:
print("Installing extensions requirements.")
extensions = next(os.walk("extensions"))[1]
for extension in extensions:
if extension in ['superbooga', 'superboogav2']: # No wheels available for requirements
continue
if initial_installation and not install_extensions:
print_big_message("Will not install extensions due to INSTALL_EXTENSIONS environment variable.")
elif initial_installation or install_extensions:
print("Installing extensions requirements.")
extensions = next(os.walk("extensions"))[1]
for extension in extensions:
if extension in ['superbooga', 'superboogav2']: # No wheels available for requirements
continue
extension_req_path = os.path.join("extensions", extension, "requirements.txt")
if os.path.exists(extension_req_path):
run_cmd("python -m pip install -r " + extension_req_path + " --upgrade", assert_success=True, environment=True)
extension_req_path = os.path.join("extensions", extension, "requirements.txt")
if os.path.exists(extension_req_path):
run_cmd("python -m pip install -r " + extension_req_path + " --upgrade", assert_success=True, environment=True)
# Detect the PyTorch version
torver = torch_version()