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