flash_attn version issue

[hwopark]

When I run the inference.ipynb, I have a problem with flash_attn issue.

ImportError: cannot import name 'flash_attn_func' from 'flash_attn' (/usr/local/lib/python3.10/dist-packages/flash_attn/init.py)

I checked the version of flash_attn is "1.0.7"

Which version of falsh_attn works?

[jshilong]

Could you provide the complete error log?