mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-09 06:45:09 +00:00
Cleanup
This commit is contained in:
@@ -40,7 +40,7 @@ def find_attention_slice_sizes(query_shape, query_element_size, query_device_typ
|
|||||||
do_split_2 = False
|
do_split_2 = False
|
||||||
do_split_3 = 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
|
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:
|
if block_size > attention_slice_rate:
|
||||||
|
|||||||
Reference in New Issue
Block a user