Upload folder using huggingface_hub

README.md

@@ -169,6 +169,13 @@ with torch.no_grad():
         torch.nn.init.normal_(p, 0, 0.1)
         print(name, p.shape)
 model.save_pretrained(save_folder)
+# print(list(model.state_dict().keys()))
+# there is some bug in model.save_pretrained... Re-save the model weights here.
+import safetensors.torch
+safetensors.torch.save_file(
+    tensors=model.state_dict(),
+    filename=f"{save_folder}/model.safetensors"
+)
 ```
 
 ### Printing the model:

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:20016f8717edbf11d8a2cfbfc65f4c83548cefddfb7f5828332d37911c44f0bb
-size 4324560
+oid sha256:39846bd3215ea587e1858ec5be2ee0b47cc7af99d941237ad52165abbfa8d18c
+size 3805256