HunyuanDiT icon indicating copy to clipboard operation
HunyuanDiT copied to clipboard

请问controlnet微调之后的.pt文件怎么转换成.safetensors呢?我在转换过程中报错了

Open Whoo-jl opened this issue 1 year ago • 1 comments

Whoo-jl avatar Jul 18 '24 11:07 Whoo-jl

试试

  1. 安装 safetensors
pip install safetensors
  1. 然后运行以下代码
import torch
from safetensors.torch import save_file

sd = torch.load("controlnet.pt")
save_file(sd, "controlnet.safetensors")

Eugeoter avatar Aug 12 '24 02:08 Eugeoter