I'd like to use your model for next word prediction, but it's hard to setup everything correctly without knowing module versions.