fix: trans_a=True also need to be modified to tl.trans()

add attention return + support eager attention or triton FA2 via config.use_flash_attn

Duplicate from zhihan1996/DNABERT-2-117M