Привет!
У меня не получилось воспроивзести ошибку, но можно попробовать отследить источник проблемы:
ошибка возникает в els = tf.gather_nd(units, idxs), где у units почему-то последняя размерность равна 0: shape = [2,21,0]. Последния размерность в shape отвечает за число NER-тэгов. Число NER-тэгов передается в конфиг файле ner_ontonotes_bert_mult.json:43: "n_tags": "#tag_vocab.len".
Проверьте содержимое файла tag.dict, который должен скачиваться командой python -m deeppavlov download ner_ontonotes_bert_mult, в нем должен быть список NER-тэгов.
Схожая ошибка обсуждалась тут: Can anyone help me find what error is this?