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 will --force-reinstall torch

Browse Source
This commit is contained in:
Lincoln Stein 2023-02-05 13:39:46 -05:00
parent fc752a4e75
commit eff7fb89d8
1 changed files with 4 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
installer/installer.py
View File

@ -130,24 +130,10 @@ class Installer:
else: else:
venv_dir = self.dest / ".venv" venv_dir = self.dest / ".venv"
# if there is an existing venv directory, then we move it out of the way
# to create a clean install
if venv_dir.exists():
backup_venv = self.dest / ".venv-backup"
print(f"Previous .venv directory detected. Renaming to {backup_venv}",end='')
if backup_venv.exists():
print(f' and removing existing {backup_venv} directory',end='')
shutil.rmtree(backup_venv)
venv_dir.replace(backup_venv)
print('')
# Prefer to copy python executables # Prefer to copy python executables
# so that updates to system python don't break InvokeAI # so that updates to system python don't break InvokeAI
try: try:
venv.create(venv_dir, with_pip=True) venv.create(venv_dir, with_pip=True)
# Because we moved away the previous .venv, the following code
# isn't strictly necessary, but keeping it here in case we decide
# it better to upgrade an existing venv rather than replace.
# If installing over an existing environment previously created with symlinks, # If installing over an existing environment previously created with symlinks,
# the executables will fail to copy. Keep symlinks in that case # the executables will fail to copy. Keep symlinks in that case
except shutil.SameFileError: except shutil.SameFileError:
@ -263,6 +249,7 @@ class InvokeAiInstance:
"--require-virtualenv", "--require-virtualenv",
"torch", "torch",
"torchvision", "torchvision",
"--force-reinstall",
"--find-links" if find_links is not None else None, "--find-links" if find_links is not None else None,
find_links, find_links,
"--extra-index-url" if extra_index_url is not None else None, "--extra-index-url" if extra_index_url is not None else None,
@ -339,6 +326,7 @@ class InvokeAiInstance:
Configure the InvokeAI runtime directory Configure the InvokeAI runtime directory
""" """
# set sys.argv to a consistent state
new_argv = [sys.argv[0]] new_argv = [sys.argv[0]]
for i in range(1,len(sys.argv)): for i in range(1,len(sys.argv)):
el = sys.argv[i] el = sys.argv[i]
@ -353,15 +341,13 @@ class InvokeAiInstance:
introduction() introduction()
from ldm.invoke.config import configure_invokeai from ldm.invoke.config import invokeai_configure
# NOTE: currently the config script does its own arg parsing! this means the command-line switches # NOTE: currently the config script does its own arg parsing! this means the command-line switches
# from the installer will also automatically propagate down to the config script. # from the installer will also automatically propagate down to the config script.
# this may change in the future with config refactoring! # this may change in the future with config refactoring!
# set sys.argv to a consistent state invokeai_configure.main()
configure_invokeai.main()
def install_user_scripts(self): def install_user_scripts(self):
""" """