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

Я могу использовать конвейер заполнения маски обнимающего лица, чтобы предсказать 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 <mask> in this sentence")

Но есть ли у кого-нибудь мнение о том, как лучше всего это сделать, если я хочу предсказать 2 замаскированных токена? например если вместо этого предложение "I am going to <mask> <mask> in this sentence"?

Если я попытаюсь поместить это точное предложение в nlp_fill, я получу сообщение об ошибке "ValueError: only one element tensors can be converted to Python scalars", поэтому оно не сработает автоматически.

Любая помощь приветствуется!