search

microsoft / UniSpeech

UniSpeech - Large Scale Self-Supervised Learning for Speech
Other
406 stars 71 forks source link

WavLM Inference Error #29

Closed bryant0918 closed 1 year ago

bryant0918 commented 2 years ago

I loaded the WaveLM Large model from the link provided

When trying to follow the code for loading the pretrained model for inference I get the following error: cfg = WavLMConfig(checkpoint['cfg']) KeyError: 'cfg'

It looks like this model does not have any 'cfg' key or 'model' key for that matter.

Sanyuan-Chen commented 2 years ago

Hi @bryant0918 ,

I just downloaded the WaveLM Large model with the Azure Storage Link and the Google Drive Link, and both models seem fine and contain the 'cfg' and 'model' keys. Could you please try again with these two links?

bryant0918 commented 1 year ago

Thank you, I originally downloaded the microsoft/wavlm-large model from huggingface and it seems to be a little different.