Removed unused argument from predictToken

the whole codebase is always predicting only the next token