test: guard against per-chunk lm_head all-gather in chunked_nll under…#6172
Draft
behroozazarkhalili wants to merge 2 commits into
Draft
test: guard against per-chunk lm_head all-gather in chunked_nll under…#6172behroozazarkhalili wants to merge 2 commits into
behroozazarkhalili wants to merge 2 commits into