Remove embeds and attempt to print output probs