update

saved/model.py

@@ -0,0 +1,21 @@
+from transformers import PreTrainedModel, PretrainedConfig
+from torch import nn
+
+class TestConfig(PretrainedConfig):
+    model_type = "test-model" 
+    
+    def __init__(self, input_dim=4, output_dim=16, **kwargs):
+        self.input_dim = input_dim
+        self.output_dim = output_dim
+        super().__init__(**kwargs)
+    
+    
+class TestModel(PreTrainedModel):
+    config_class = TestConfig
+    
+    def __init__(self, config):
+        super().__init__(config)
+        self.layer = nn.Linear(config.input_dim, config.output_dim)
+        
+    def forward(self, input):
+        return self.layer(input)

{pytorch_model.bin → saved}/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:218cdc4788564080b5cb32506374a2dc69c766c1313c96eaf08aa34f922f20f3
+oid sha256:6129e117260ef9958a341f3a738c069b1dbdde687e29b8de423475dd5c3e631b
 size 496