File size: 425 Bytes
b344ca4 |
1 2 3 4 5 6 7 8 9 10 11 12 |
import torch
from safetensors.torch import load_file, save_file
file_path = "./hunyuan_dit_1.1.safetensors"
model_path = ("./mp_rank_00_model_states.pt")
loaded = load_file(file_path)
sd = torch.load(model_path, map_location=lambda storage, loc: storage)
for i in sd["module"]:
loaded["model."+str(i)] = sd["module"][i]
save_file(loaded, "Freeway_Animation_HunYuan_Demo_comfyui.safetensors")
# manual surgery |