pytext/models/doc_model.py
pytext/torchscript/module.py

pytext/models/word_model.py
pytext/torchscript/module.py