fix flash_attn import logic to comply with huggingface import check c2ced98 verified smg478 commited on Oct 31