SD1 uses a head dimension that is not supported by the old version of Flash Attention but is supported by this version.