It would be nice to be able to generate e.g. bocchi: ryo with the colon when given only bocchi without a colon as prompt. This was one of the goals of #45. It doesn't work now, probably due to the extra space added at the end of the last token by brain.Tokens.