Update README.md

README.md

@@ -1,8 +1,16 @@
 ```python
+import json
 import onnx
 import torch
 from safetensors.torch import save_file
 from huggingface_hub import hf_hub_download
-save_file({initializer.name : torch.tensor(onnx.numpy_helper.to_array(initializer)) for initializer in onnx.load(hf_hub_download("KittenML/kitten-tts-nano-0.8-int8","kitten_tts_nano_v0_8.onnx")).graph.initializer}, "model.safetensors")
+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")
 
 ```