Move flash_attn assert from __init__ into calling func (#32) ad85cab verified nguyenbh rogerxfeng8 commited on Sep 12