Update launch.py
This commit is contained in:
parent
cfd772682e
commit
9102e404b2
|
|
@ -29,7 +29,7 @@ TRY_INSTALL_XFORMERS = False
|
|||
def prepare_environment():
|
||||
torch_index_url = os.environ.get('TORCH_INDEX_URL', "https://download.pytorch.org/whl/cu121")
|
||||
torch_command = os.environ.get('TORCH_COMMAND',
|
||||
f"pip install torch==2.1.0 torchvision==0.16.0 --extra-index-url {torch_index_url}")
|
||||
f"pip install torch==2.7.1+cu128 orchvision==0.22.1+cu128 --extra-index-url {torch_index_url}")
|
||||
requirements_file = os.environ.get('REQS_FILE', "requirements_versions.txt")
|
||||
|
||||
print(f"Python {sys.version}")
|
||||
|
|
|
|||
Loading…
Reference in New Issue