tkob-vh

icon indicating a website link https://tkob-vh.github.io

icon company Huazhong University of Science and Technology icon location Wuhan

Results 1 comments of tkob-vh

AttributeError: module 'transformer_engine' has no attribute 'pytorch'

Try replace the `flash-attn-3` / `flash_attn_3` with `flash-attn` / `flash_attn` in `transformer_engine/pytorch/attention/dot_product_attention/backends.py` ?