Revert "Optionally use flash-attn's CE loss for metrics (#3394)"#3408
Merged
snarayan21 merged 1 commit intodevfrom saaketh/revert_fa_ce_lossJun 18, 2024
+4-118
flash-attn's CE loss for metrics (#3394)"#3408