mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add INSTALL_EXTENSIONS environment variable

Browse source
This commit is contained in:
oobabooga 2023-09-25 13:12:35 -07:00
parent 31f2815a04
commit 44438c60e5
2 changed files with 16 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

24
one_click.py
View file

@ -183,16 +183,22 @@ def update_requirements(initial_installation=False):
run_cmd("git pull --autostash", assert_success=True, environment=True)
# Initial installation only: install the extensions requirements
if initial_installation:
extensions = next(os.walk("extensions"))[1]
for extension in extensions:
if extension in ['superbooga']: # No wheels available for requirements
continue
# 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']: # 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()