# Visualize the top scoring tokens after each layer
output.layer_predictions(position=6)