ImportError: cannot import name 'AutoModelForMaskedLM'

marcotcr / checklist

Beyond Accuracy: Behavioral Testing of NLP models with CheckList

MIT License

2.01k stars 204 forks source link

Closed feng-1985 closed 3 years ago

feng-1985 commented 3 years ago

env: transformers-2.8.0 torch-1.7.0

marcotcr commented 3 years ago

You need to update transformers, that version doesn't have AutoModelForMaskedLM