Try to separate the tokenization from the model's forward pass.