got rid of end_idxs since get_predictions no longer needs this
got rid of end_idxs since get_predictions no longer needs this

facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI,facebookresearch/ParlAI