nguyenbh's picture rogerxfeng8's picture
Move flash_attn assert from __init__ into calling func (#32)
ad85cab verified