Использование модели T5 с конвейером заполнения маски - PullRequest
1 голос
/ 24 апреля 2020

Кто-нибудь знает, возможно ли использовать модель T5 с обтягивающим конвейером для маскировки лица? Ниже показано, как вы можете сделать это, используя модель по умолчанию, но я не могу понять, как это сделать, используя модель T5 специально?

from transformers import pipeline
nlp_fill = pipeline('fill-mask')
nlp_fill('Hugging Face is a French company based in ' + nlp_fill.tokenizer.mask_token)

Попытка сделать это, например, приводит к ошибке «TypeError: должно быть str, а не NoneType», поскольку nlp_fill.tokenizer.mask_token имеет значение None.

nlp_fill = pipeline('fill-mask',model="t5-base", tokenizer="t5-base")
nlp_fill('Hugging Face is a French company based in ' + nlp_fill.tokenizer.mask_token)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...