From f7d83b72e0d4dd27ce6e54ef77dfb2ae4cb0edcd Mon Sep 17 00:00:00 2001 From: zhangp365 <144313702+zhangp365@users.noreply.github.com> Date: Mon, 23 Dec 2024 12:44:20 +0800 Subject: [PATCH] fixed a bug in ldm/pixart/blocks.py (#6158) --- comfy/ldm/pixart/blocks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/pixart/blocks.py b/comfy/ldm/pixart/blocks.py index 48b27008..967a224a 100644 --- a/comfy/ldm/pixart/blocks.py +++ b/comfy/ldm/pixart/blocks.py @@ -12,7 +12,7 @@ from comfy.ldm.modules.attention import optimized_attention if model_management.xformers_enabled(): import xformers.ops - if int((xformers.__version__).split(".")[2]) >= 28: + if int((xformers.__version__).split(".")[2].split("+")[0]) >= 28: block_diagonal_mask_from_seqlens = xformers.ops.fmha.attn_bias.BlockDiagonalMask.from_seqlens else: block_diagonal_mask_from_seqlens = xformers.ops.fmha.BlockDiagonalMask.from_seqlens