From dd4412734e32895879f3f3cbfb5a42715d43c7be Mon Sep 17 00:00:00 2001 From: Disty0 Date: Mon, 7 Aug 2023 11:32:00 +0300 Subject: [PATCH] IPEX add torchvision on Windows --- installer.py | 2 +- modules/ipex_specific/diffusers.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/installer.py b/installer.py index 644b6b4b8..67390b382 100644 --- a/installer.py +++ b/installer.py @@ -328,7 +328,7 @@ def check_torch(): torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.0.1a0 torchvision==0.15.2a0 intel_extension_for_pytorch==2.0.110+xpu -f https://developer.intel.com/ipex-whl-stable-xpu') os.environ.setdefault('TENSORFLOW_PACKAGE', 'tensorflow==2.13.0 intel-extension-for-tensorflow[gpu]') else: - torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.0.0a0 intel_extension_for_pytorch==2.0.110+gitba7f6c1 -f https://developer.intel.com/ipex-whl-stable-xpu') + torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.0.0a0 torchvision==0.15.2a0 intel_extension_for_pytorch==2.0.110+gitba7f6c1 -f https://developer.intel.com/ipex-whl-stable-xpu') else: machine = platform.machine() if sys.platform == 'darwin': diff --git a/modules/ipex_specific/diffusers.py b/modules/ipex_specific/diffusers.py index 2312fcf8b..f8fab5a0f 100644 --- a/modules/ipex_specific/diffusers.py +++ b/modules/ipex_specific/diffusers.py @@ -58,8 +58,12 @@ class SlicedAttnProcessor: if block_size >= 4000: do_split_2 = True #Find something divisible with the query_tokens + sanity_check = 0 while ((self.slice_size * split_2_slice_size * shape_three) / 1024 * block_multiply) > 4000: split_2_slice_size = split_2_slice_size // 2 + sanity_check = sanity_check + 1 + if sanity_check >= 128: + break else: do_split_2 = False