From b9d21811927092e33f4e74fa33809bf2cf11ceca Mon Sep 17 00:00:00 2001 From: Disty0 Date: Tue, 2 Jan 2024 11:51:29 +0300 Subject: [PATCH] Cleanup --- library/ipex/diffusers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/ipex/diffusers.py b/library/ipex/diffusers.py index 56528551..47b0375a 100644 --- a/library/ipex/diffusers.py +++ b/library/ipex/diffusers.py @@ -40,7 +40,7 @@ def find_attention_slice_sizes(query_shape, query_element_size, query_device_typ do_split_2 = False do_split_3 = False - if query_device_type != "xpu" + if query_device_type != "xpu": return do_split, do_split_2, do_split_3, split_slice_size, split_2_slice_size, split_3_slice_size if block_size > attention_slice_rate: