298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed a bug in ldm/pixart/blocks.py

Browse Source
This commit is contained in:
zhangp365 2024-12-22 20:20:48 +08:00 committed by GitHub
parent 57f330caf9
commit ccdd4a64ea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/pixart/blocks.py
View File

@ -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