IPEX add torchvision on Windows

2026-09-19 01:04:32 +02:00 · 2023-08-07 11:32:00 +03:00
parent dc739b9f50
commit dd4412734e
2 changed files with 5 additions and 1 deletions
@@ -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