ComfyUI_ExtraModels icon indicating copy to clipboard operation
ComfyUI_ExtraModels copied to clipboard

need fix BlockDiagonalMask error for xformer 0.0.28.post1

Open SaiZyca opened this issue 1 year ago • 0 comments

PixArt and SANA need fix code for newer xformers


  if (xformers.__version__).split(".")[2] >= "28":
      attn_bias = xformers.ops.fmha.attn_bias.BlockDiagonalMask.from_seqlens([N] * B, mask)
  else:
      attn_bias = xformers.ops.fmha.BlockDiagonalMask.from_seqlens([N] * B, mask)

SaiZyca avatar Dec 09 '24 07:12 SaiZyca