And inflash-attn 1.0.4you should have the class, see here:https://github.com/Dao-AILab/flash-attention/blob/v1.0.4/flash_attn/flash_attention.py#L74 Therefore, I think this is still likely an installation issue. Could you try in your environment, python -c"import flash_attn"python -c...