guoyww/AnimateDiff · Error when running locally

When I try to run locally I get the following error:

Traceback (most recent call last):
File "path/to/AnimateDiff/app.py", line 577, in
controller = AnimateController()
File "path/to/AnimateDiff/app.py", line 472, in init
self.update_base_model(self.base_model_list[0])
File "path/to/AnimateDiff/app.py", line 498, in update_base_model
converted_vae_checkpoint = convert_ldm_vae_checkpoint(base_model_state_dict, self.vae.config)
File "path/to/AnimateDiff/animatediff/utils/convert_from_ckpt.py", line 574, in convert_ldm_vae_checkpoint
new_checkpoint["encoder.conv_in.weight"] = vae_state_dict["encoder.conv_in.weight"]
KeyError: 'encoder.conv_in.weight'

Trying to print a few steps along the way to see what could be happening I have found that in convert_ldm_vae_checkpoint, in keys which contains all checkpoint keys there are none that start with vae_key "first_stage_model".

Does anyone have any idea what could be happening?

Thanks