add attention return + support eager attention or triton FA2 via config.use_flash_attn 42c0941 verified Taykhoom commited on 1 day ago