The model's pipeline returns an error

#2
by Eduarda - opened

Just by using the default code provided in the model's page, the pipeline returns the error KeyError: 'logits'. The pipeline is loaded fine in thepipe variable but, when we test the pipeline with a string, the snippet pipe('Tinha uma [MASK] no meio do caminho.') returns an error. I have tried multiple environments, and devices and acquired the same result.

import torch
from transformers import AutoTokenizer
from transformers import AutoModelForPreTraining
from transformers import AutoModel
​
model = AutoModelForPreTraining.from_pretrained('neuralmind/bert-base-portuguese-cased')
tokenizer = AutoTokenizer.from_pretrained('neuralmind/bert-base-portuguese-cased', do_lower_case=False)
​
from transformers import pipeline
​
pipe = pipeline('fill-mask', model=model, tokenizer=tokenizer)
​
pipe('Tinha uma [MASK] no meio do caminho.')
Eduarda changed discussion status to closed
Eduarda changed discussion status to open

When the model is downloaded directly from the pipeline function, the error is not displayed and the pipe returns the results for the [MASK] normally.

pipe = pipeline('fill-mask', model='neuralmind/bert-base-portuguese-cased')

pipe("Tinha uma [MASK] no meio do caminho.")

Does any one know the reason for this?

I got the same error, trying in Colab.

got the same error, following. also, Eduarda's workaround also worked for me

Sign up or log in to comment