Mirrors/AnimateDiff
1
0
Fork 0
mirror of https://github.com/guoyww/AnimateDiff.git synced 2026-04-03 17:56:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

add dummy key

Browse Source
This commit is contained in:
Yuwei Guo
2023-12-16 12:50:16 +08:00
parent 57e7d14ede
commit 0e9ad276e7
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
scripts/animate.py
View File

@@ -69,6 +69,7 @@ def main(args):
print(f"loading controlnet checkpoint from {model_config.controlnet_path} ...")
controlnet_state_dict = torch.load(model_config.controlnet_path, map_location="cpu")
controlnet_state_dict = controlnet_state_dict["controlnet"] if "controlnet" in controlnet_state_dict else controlnet_state_dict
controlnet_state_dict.pop("animatediff_config", "")
controlnet.load_state_dict(controlnet_state_dict)
controlnet.cuda()