yizhilll/MERT

How to convert a fairseq checkpoint to a huggingface checkpoint

z592694590 opened this issue · 1 comments

z592694590 commented

Hi, thanks for this great job.
I have trained a mert model by your repo. But I can't use the fairseq checkpoint to infer. Could you provide some help？I see the demo of infer is based on huggingface. How can I convert a fairseq checkpoint to a huggingface checkpoint.

yizhilll commented

Hi, I've uploaded scripts for conversion. Please check with the scripts:

bash scripts/convert_HF_script.sh default mert config_mert_base [/absolute/path/to/a/fairseq/checkpoint.pt]
  • 👍1