transformers

e7337ee7 - Pass num_items_in_batch directly to loss computation (#36753)

Pass num_items_in_batch directly to loss computation (#36753) * Pass num_items_in_batch directly to loss computation * use self loss instead * fix loss kwrgs * fix vocab size