fix imports to work with transformers 0.4