fix compatibility with transformers-0.4.0.0