Optionally use flash-attn
's CE loss for metrics
#6019
This job succeeded
Loading
flash-attn
's CE loss for metrics
#6019