diff options
| author | Sakura-Luna <53183413+Sakura-Luna@users.noreply.github.com> | 2023-05-14 14:29:28 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-14 14:29:28 +0000 |
| commit | f29c41bf6dd04a721e1dc0a162b042cad269f247 (patch) | |
| tree | 9f688b643d557aa82498ed785a59d60972a0f6f5 /launch.py | |
| parent | ef046fae39868d25be9bc1a6b7863bcd22e550c2 (diff) | |
| download | stable-diffusion-webui-gfx803-f29c41bf6dd04a721e1dc0a162b042cad269f247.tar.gz stable-diffusion-webui-gfx803-f29c41bf6dd04a721e1dc0a162b042cad269f247.tar.bz2 stable-diffusion-webui-gfx803-f29c41bf6dd04a721e1dc0a162b042cad269f247.zip | |
Modify pytorch command
Diffstat (limited to 'launch.py')
| -rw-r--r-- | launch.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -237,7 +237,7 @@ def run_extensions_installers(settings_file): def prepare_environment():
- torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118")
+ torch_command = os.environ.get('TORCH_COMMAND', "pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --index-url https://download.pytorch.org/whl/cu118")
requirements_file = os.environ.get('REQS_FILE', "requirements_versions.txt")
xformers_package = os.environ.get('XFORMERS_PACKAGE', 'xformers==0.0.17')
|