search

PrimerAI / blanc

Human-free quality estimation of document summaries
MIT License
94 stars 11 forks source link

Migration to Transformers version 4.x #14

Closed zakhan4 closed 3 years ago

zakhan4 commented 3 years ago

Hi,

It would be great if the current blanc implementation could be adjusted to work with Transformer Version 4.x, as it is failing due to some breaking changes!

https://huggingface.co/transformers/migration.html

Thanks!

OlegVasilyev4096 commented 3 years ago

Thanks! We'll fix this

OlegVasilyev4096 commented 3 years ago

Seems deprecation of 'masked_lm_labels' was the only issue, other breaking changes were already in. Works for transformers 4.x including 4.6.1. Let us know if anything else, thanks.