it doesnt generate text though, only probabilities