For the compatibility with the latest transformers lib. a93b171 verified ZhengPeng7 commited on Sep 12