import torch
from safetensors.torch import load_file

# 1. 载入 safetensors 权重文件
safetensor_path = "diffusion_pytorch_model.safetensors"
state_dict = load_file(safetensor_path)

# 2. 保存为 PyTorch 的 .pth 格式
torch.save(state_dict, "Wan2.1_VAE.pth")

print("转换完成：Wan2.1_VAE.pth")