predict_mask.py运行失败

tongchangD / bert_for_corrector

基于bert进行中文文本纠错

225 stars 49 forks source link

predict_mask.py运行失败 #20

Open Hachiiiiii opened 2 years ago

Hachiiiiii commented 2 years ago

总是会报这个错误 ValueError: Could not load model ./data/bert_models/chinese_finetuned_lm/pytorch_model.bin with any of the following classes: (<class 'transformers.models.auto.modeling_auto.AutoModelForMaskedLM'>,). 请问您知道是什么原因吗

tongchangD commented 2 years ago

你看看tensformer版本,这是我几年前弄的一个小demo.可能版本不对应了,你查查版本