search

feizc / Dimba

Transformer-Mamba Diffusion Models
75 stars 5 forks source link

VERSION OF xformers? #5

Open zefengzhang123 opened 3 months ago

zefengzhang123 commented 3 months ago

AttributeError: module 'xformers.ops.fmha' has no attribute 'BlockDiagonalMask'

feizc commented 3 months ago

Hi, you can install the xformers with pip install xformers==0.0.23.post1 or download whl from https://download.pytorch.org/whl/xformers/ for 0.0.23.post1 version.