search

luosiallen / Diff-Foley

Diff-Foley: Synchronized Video-to-Audio Synthesis with Latent Diffusion Models
Apache License 2.0
147 stars 15 forks source link

KeyError in LDM Model #3

Closed aashishrai3799 closed 10 months ago

aashishrai3799 commented 10 months ago

Hi, thanks again for releasing this amazing work.

I'm facing an issue while loading LDM from checkpoint (latent_diffusion_model = load_model_from_config(config, ldm_ckpt_path)). It's unable to find the "state_dict" key in the pl_sd in (sd = pl_sd["state_dict"]).

Here's the error: Traceback (most recent call last): File "", line 1, in KeyError: 'state_dict'

Your help will be greatly appreciated.

luosiallen commented 10 months ago

Thanks @aashishrai3799 , just re-upload the LDM checkpoint again in huggingface. Can try it again!