FileNotFoundError: [Errno 2] No such file or directory: '/home/karen/.cache/audioldm/audioldm-s-full.ckpt'

[VinceAlle]

When I follow the README.md to self-train this model, there is an error stopping the train
pretrained_model_name = "audioldm-s-full" vae, stft = build_pretrained_models(pretrained_model_name) def build_pretrained_models(name): checkpoint = torch.load(get_metadata()[name]["path"], map_location="cpu",weights_only=True)
FileNotFoundError: [Errno 2] No such file or directory: '/home/karen/.cache/audioldm/audioldm-s-full.ckpt'
But when I check the folder '/home/karen/.cache/audioldm/', there is a file named 'audioldm-s-full' instead of 'audioldm-s-full.ckpt'
So what I need to do is just change the 'tango/audioldm/utils.py' in line 209 from "audioldm-s-full.ckpt" to "audioldm-s-full". After that, the Error not show again.
Hope the owner can fix it.

[aixingxy]

来函妥收。