chore: add new safetensors model

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9bbcbf73561f6bc5d0a17ea6a2081feed2d1304e87602d8c502d9a5c4bd85576
+size 16

src/init_onnx.py

@@ -1,6 +1,6 @@
 import torch
 import torch.nn as nn
-import torch.onnx
+
 class BaseModel(nn.Module):
     def __init__(self):
         super(BaseModel, self).__init__()
@@ -13,6 +13,12 @@ model = BaseModel()
 
 dummy_input = torch.tensor([[[[1, 2, 3], [4, 5, 6], [7, 8, 9]]]], dtype=torch.float32)
 
+safetensors_file_path = "model.safetensors"
+
+from safetensors.torch import save_file
+save_file(model.state_dict(), 'model.safetensors')
+
+import torch.onnx
 onnx_file_path = "model.onnx"
 torch.onnx.export(model, dummy_input, onnx_file_path,
                   input_names=['input'], output_names=['output'],