ModuleNotFoundError: No module named 'onmt'

[stalevna]

Hi,
I run into an error about onmt

---

ModuleNotFoundError Traceback (most recent call last)
in
5 import torch.nn.functional as F
6 from collections import OrderedDict
----> 7 from onmt.utils.misc import sequence_mask
8 from model_autopst import Generator_2 as Predictor
9 from hparams_autopst import hparams

ModuleNotFoundError: No module named 'onmt'

but I see folder onmt_modules exists

then I install onmt(pip install onmt) and notice it's installing torch 1.3.0 although the requirements say that PyTorch == v1.6.0

Could you help me with this issue? What is the best approach to solve this?

[auspicious3000]

from onmt_modules.misc import sequence_mask