transformers

d996024a - Use compute_loss in prediction_step (#9935)