When doing model prediction, how to get the next-to-last layer emeddings at the same time? #271