update tok2vec.py