Don’t want to bump the thread too much, but like mentioned before, pickling the tokenizer object works when I load the model in another session. I just want to know if there’s a better way for this, such as having a layer in the model definition that does all that without having to load the tokenizer every time.