From 09ad0e043ab1c2d8c3a071815c8937458417cfe0 Mon Sep 17 00:00:00 2001 From: eyecreate Date: Tue, 9 May 2023 17:09:58 -0400 Subject: [PATCH] Update installer.py to not be tied to quickly changing intel torch versions If you leave off the git revision, you can still pull in the latest torch build from intel. If you don't do this, intel users will have the install process fail due to new/different versions of pytorch intel pushes. --- installer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/installer.py b/installer.py index 3f5d1b892..b0fab68c6 100644 --- 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()