fairseq-detect-hallucination icon indicating copy to clipboard operation
fairseq-detect-hallucination copied to clipboard

Published 20 hours ago • verified publisher icon violet-zct

Regarding "predict_hallucination_mt.py"

Open vijini opened this issue 2 years ago • 1 comments

I tried to run the predict_hallucination_mt.py file by downloading the pre-trained xsum model. But when, xlmr = XLMRModel.from_pretrained( model_path, checkpoint_file='checkpoint.pt' data_name_or_path=datapath ) is called, a key error occurs as follows, KeyError: 'token_prediction'

Can you please advice me on this issue?

vijini avatar May 19 '22 11:05 vijini

@vijini Fisrt, create a new conda environment or pipenv for this repo . Then, follow the README.md or description in this link https://github.com/violet-zct/fairseq-detect-hallucination/issues/1#issuecomment-1073051054

Lyshc-cgc avatar May 31 '22 07:05 Lyshc-cgc