Delete saved

saved/config.json

@@ -1,14 +0,0 @@
-{
-  "architectures": [
-    "TestModel"
-  ],
-  "auto_map": {
-    "AutoConfig": "model.TestConfig",
-    "AutoModel": "model.TestModel"
-  },
-  "input_dim": 4,
-  "model_type": "test-model",
-  "output_dim": 16,
-  "torch_dtype": "float32",
-  "transformers_version": "4.54.1"
-}

saved/model.py

@@ -1,21 +0,0 @@
-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)

saved/model.safetensors

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