From 4ba9f0c8dc128db4ce9b56a86360759cb0fbef66 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Tue, 9 May 2023 12:22:52 -0400 Subject: [PATCH] pin torch version --- extensions-builtin/stable-diffusion-webui-images-browser | 2 +- installer.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/extensions-builtin/stable-diffusion-webui-images-browser b/extensions-builtin/stable-diffusion-webui-images-browser index e535ea6ae..c751a9eee 160000 --- a/extensions-builtin/stable-diffusion-webui-images-browser +++ b/extensions-builtin/stable-diffusion-webui-images-browser @@ -1 +1 @@ -Subproject commit e535ea6aea2356a13da3e0452b92b3d68861f539 +Subproject commit c751a9eeedef738fd0a731db1dd65690690b6161 diff --git a/installer.py b/installer.py index 130ad6de9..3f5d1b892 100644 --- a/installer.py +++ b/installer.py @@ -192,12 +192,12 @@ def check_python(): def check_torch(): if shutil.which('nvidia-smi') is not None or os.path.exists(os.path.join(os.environ.get('SystemRoot') or r'C:\Windows', 'System32', 'nvidia-smi.exe')): log.info('nVidia CUDA toolkit detected') - torch_command = os.environ.get('TORCH_COMMAND', 'torch torchaudio torchvision --index-url https://download.pytorch.org/whl/cu118') + torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.0.0 torchaudio torchvision --index-url https://download.pytorch.org/whl/cu118') xformers_package = os.environ.get('XFORMERS_PACKAGE', 'xformers==0.0.17' if opts.get('cross_attention_optimization', '') == 'xFormers' else 'none') elif shutil.which('rocminfo') is not None or os.path.exists('/opt/rocm/bin/rocminfo'): log.info('AMD ROCm toolkit detected') os.environ.setdefault('HSA_OVERRIDE_GFX_VERSION', '10.3.0') - torch_command = os.environ.get('TORCH_COMMAND', 'torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2') + torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.0.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2') xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none') elif shutil.which('sycl-ls') is not None or os.path.exists('/opt/intel/oneapi') or args.use_ipex: log.info('Intel OneAPI Toolkit detected')