Se você já possui modelos treinados em formato .pt ou .pth, é possível convertê-los para SafeTensor com poucas linhas de código:
import torch
from safetensors.torch import save_file
state_dict = torch.load(“modelo_antigo.pt”, map_location=”cpu”)
save_file(state_dict, “modelo_convertido.safetensors”)
Quer saber mais sobre O que é safetensor, Leia nosso artigo completo aqui: