commit to merge
commit to merge
fixes for language model to be able to generate text on different training data.