Merge pull request vladmandic#842 from eyecreate/patch-1

Update installer.py to not be tied to quickly changing intel torch ve…
Artheriax · May 9, 2023 · faf2cd9 · faf2cd9
2 parents 5bd863f + 09ad0e0
commit faf2cd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/installer.py b/installer.py
@@ -201,7 +201,7 @@ def check_torch():
         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')
-        torch_command = os.environ.get('TORCH_COMMAND', 'torch==1.13.0a0+git6c9b55e torchvision==0.14.1a0 intel_extension_for_pytorch==1.13.120+xpu -f https://developer.intel.com/ipex-whl-stable-xpu')
+        torch_command = os.environ.get('TORCH_COMMAND', 'torch==1.13.0a0 torchvision==0.14.1a0 intel_extension_for_pytorch==1.13.120+xpu -f https://developer.intel.com/ipex-whl-stable-xpu')
         xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none')
     else:
         machine = platform.machine()