kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Safetensor loading in AnimateDiff conversion scripts (#7764)

Browse Source 
* update

* update
This commit is contained in:
Dhruv Nair 2024-04-29 17:36:50 +05:30 committed by GitHub
parent a38dd79512
commit eb96ff0d59
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/convert_animatediff_motion_lora_to_diffusers.py
View File

@ -1,7 +1,7 @@
import argparse
import torch
from safetensors.torch import save_file
from safetensors.torch import load_file, save_file
def convert_motion_module(original_state_dict):
@ -34,7 +34,10 @@ def get_args():
if __name__ == "__main__":
args = get_args()
state_dict = torch.load(args.ckpt_path, map_location="cpu")
if args.ckpt_path.endswith(".safetensors"):
state_dict = load_file(args.ckpt_path)
else:
state_dict = torch.load(args.ckpt_path, map_location="cpu")
if "state_dict" in state_dict.keys():
state_dict = state_dict["state_dict"]

7
scripts/convert_animatediff_motion_module_to_diffusers.py
View File

@ -1,6 +1,7 @@
import argparse
import torch
from safetensors.torch import load_file
from diffusers import MotionAdapter
@ -38,7 +39,11 @@ def get_args():
if __name__ == "__main__":
args = get_args()
state_dict = torch.load(args.ckpt_path, map_location="cpu")
if args.ckpt_path.endswith(".safetensors"):
state_dict = load_file(args.ckpt_path)
else:
state_dict = torch.load(args.ckpt_path, map_location="cpu")
if "state_dict" in state_dict.keys():
state_dict = state_dict["state_dict"]