chore: ad base model

config.json

@@ -0,0 +1,4 @@
+{
+  "model_type": "onnx-base",
+  "model_name_or_path": "model.onnx"
+}

model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:340fee2e31d28c3d73f480c802a5c10f81bfe067ec4658c8dade05943dcab2b4
+size 271

requirements.txt

@@ -0,0 +1 @@
+torch

src/init_model.py

@@ -0,0 +1,18 @@
+import torch
+import torch.nn as nn
+import torch.onnx
+
+# Define a simple model
+class SimpleModel(nn.Module):
+    def __init__(self):
+        super(SimpleModel, self).__init__()
+        self.fc = nn.Linear(10, 1)
+
+    def forward(self, x):
+        return self.fc(x)
+
+# Instantiate and export the model
+model = SimpleModel()
+dummy_input = torch.randn(1, 10)
+onnx_path = "../model.onnx"
+torch.onnx.export(model, dummy_input, onnx_path, input_names=['input'], output_names=['output'])