KeyError: 'generator'

[Akira4ever]

KeyError: 'generator'

[cndn]

Hey @Akira4ever - The code assumes vocoder checkpoint has states stored in 'generator' field. Could you print out checkpoint.keys() in the line throwing error above and see what fields it has in your checkpoint?

Looking at what you've done, one worry I have is you might hardcode file path of the UnitY model in fairseq2/models/utils/model_loader.py but the model loader file is shared with vocoder. If you'd like loading from local path instead of downloading on the fly, you could also update the yaml file with local paths as #50

[Akira4ever]

Thanks! I indeed changed file path of the model and maybe that's the reason. : (

[Akira4ever]

you might hardcode file path of the UnitY model in fairseq2/models/utils/model_loader.py but the model loader file is shared with vocoder.

And it's because I didn't download the vocoder model from huggingface...But #50 also helps me. Thank you!