Ошибка в версии DeepPavlov-1.0.0rc1

Пытаюсь использовать DeepPavlov-1.0.0rc1для работы с CUDA 11
с Python-3.6.8, torch 1.10.2.
При использовании модели ner_rus_bert , получаю -

File “…/DeepPavlov-1.0.0rc1/deeppavlov/core/models/torch_model.py”, line 178, in load
self.model.module.load_state_dict(model_state)
File “…/python3.6.8.env/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 1178, in getattr
type(self).name, name))
AttributeError: ‘BertForTokenClassification’ object has no attribute ‘module’