Лучший способ использовать заполнение маски обнимающего лица для более чем 1 жетона маски одновременно. ⇐ Python

[Anonymous]

Я могу использовать конвейер заполнения маски обнимающего лица, чтобы предсказать 1 замаскированный токен в предложении, используя следующее:

Код: Выделить всё

!pip install -q transformers
from __future__ import print_function
import ipywidgets as widgets
from transformers import pipeline

nlp_fill = pipeline('fill-mask')
nlp_fill("I am going to guess  in this sentence")

Но есть ли у кого-нибудь мнение о том, как лучше всего это сделать, если я хочу предсказать 2 замаскированных токена? например если вместо этого предложение будет «Я собираюсь использовать в этом предложении»?

Если я попытаюсь поместить это точное предложение в nlp_fill, я получу ошибку «ValueError: только один тензор элемента может быть преобразован в скаляры Python», так что это не так. не работает автоматически.

Будем благодарны за любую помощь!

Подробнее здесь: https://stackoverflow.com/questions/609 ... token-at-a