From ccdd4a64ea1a3fe63d2dbce68a78d549d233290c Mon Sep 17 00:00:00 2001 From: zhangp365 <144313702+zhangp365@users.noreply.github.com> Date: Sun, 22 Dec 2024 20:20:48 +0800 Subject: [PATCH] fixed a bug in ldm/pixart/blocks.py --- 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 48b27008c..967a224a3 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