shethjenil/KittenTTS

import json
import onnx
import torch
from safetensors.torch import save_file
from huggingface_hub import hf_hub_download
state_dict = {}
state_dict_info = {}
for i in onnx.load(hf_hub_download("KittenML/kitten-tts-nano-0.8-int8","kitten_tts_nano_v0_8.onnx")).graph.initializer:
    p = onnx.numpy_helper.to_array(i)
    state_dict_info[i.name] = list(p.shape)
    state_dict[i.name] = torch.tensor(p)
json.dump(state_dict_info,open("KittenTTS.json","w"))
save_file(state_dict, "model.safetensors")