Upload folder using huggingface_hub
Browse files- model.safetensors +3 -0
- to_safetensors.py +5 -0
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ef5ab55b5daa68066891c2991cfe1628f76581bf0106deedb4b2fa907ae4dead
|
| 3 |
+
size 2635388
|
to_safetensors.py
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import torch
|
| 2 |
+
from safetensors.torch import save_file
|
| 3 |
+
|
| 4 |
+
model = torch.load("model.pth", weights_only=False)
|
| 5 |
+
save_file(model.state_dict(), "model.safetensors")
|