Fix code to comport with newer Transformers library 6c59ef9 verified ctranslate2-4you commited on Feb 15