transformers
Fix flaky `test_causal_lm_loss_honors_shift_labels`
#50533

Merged

Fix flaky `test_causal_lm_loss_honors_shift_labels` #50533

ydshieh merged 3 commits into main from fix_flaky_shift_labels_loss_test

Fix flaky `test_causal_lm_loss_honors_shift_labels`

7d05cef6

Address review: probe logits without labels, drop skip guard

9a138d83

Select the logits the loss is computed on before picking `shift_labels`

4bb71b39

ydshieh merged 78a4dafe into main 2 days ago

ydshieh deleted the fix_flaky_shift_labels_loss_test branch 2 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone