I’m trying to import ElectraForMultipleChoice from transformers, but I keep getting this error. I also attempted to install the most recent version of transformers with ‘pip install transformers --upgrade,’ but that did not work.
from transformers import AutoTokenizer, ElectraForMultipleChoice
import torch
​
tokenizer = AutoTokenizer.from_pretrained("google/electra-small-discriminator")
model = ElectraForMultipleChoice.from_pretrained("google/electra-small-discriminator")
---------------------------------------------------------------------------
ImportError Traceback (most recent call last)
<ipython-input-43-eb23a2fcf7a6> in <module>
----> 1 from transformers import AutoTokenizer, ElectraForMultipleChoice
2 import torch
3
4 tokenizer = AutoTokenizer.from_pretrained("google/electra-small-discriminator")
5 model = ElectraForMultipleChoice.from_pretrained("google/electra-small-discriminator")
ImportError: cannot import name 'ElectraForMultipleChoice' from 'transformers' (/opt/conda/lib/python3.7/site-packages/transformers/__init__.py)