Использование векторизации RUBERT

Добрый день! Подскажите, пожалуйста, как использовать RuBert для векторизации слов. В стандарных конфигурациях RuBert входит в состав полной модели, к примеру NerBert. После RuBert планирую использовать модель из сторонней библиотеки.

На данный момент есть пулл реквест с данным функционалом. https://github.com/deepmipt/DeepPavlov/pull/1039. Можете поставить эту ветку командой: pip install git+ssh://git@github.com/deepmipt/deeppavlov.git@feat/bert-embedder
Использовать можно так:

m = build_model(configs.embedder.bert_embedder)
embeddings = m(['hi, i want my embedding', 'hi'])
1 Like

Спасибо! Поставил это пулл реквест. Вылетает ошибка!

починил. Установил bert_dp из другого конфигурационного файла